如何基于NumPy数组两列生成含重复元素的一维ndarray

嗨，这事儿用NumPy自带的np.repeat()方法就能轻松搞定，完全不用手动写循环，效率还高！

步骤1：构造示例数组 #

先把你给出的二维数组用NumPy构造出来：

import numpy as np

# 对应你给出的key和num_of_instances
arr = np.array([[99, 2],
                [88, 3],
                [77, 1]])

步骤2：用np.repeat实现展平 #

直接调用np.repeat()，第一个参数传入key列，第二个参数传入对应的实例数列：

flattened_arr = np.repeat(arr[:, 0], arr[:, 1])
print(flattened_arr)

运行后输出就是你想要的结果：

[99 99 88 88 88 77]

简单解释一下 #

• arr[:, 0]：取数组中所有行的第一列，也就是我们要重复的key值数组[99,88,77]
• arr[:, 1]：取数组中所有行的第二列，也就是每个key对应的重复次数数组[2,3,1]
• np.repeat()会自动把第一个数组里的每个元素，按照第二个数组对应位置的次数重复，正好满足你的需求~

内容的提问来源于stack exchange，提问作者Yanirmr