代码之家 › 专栏 › 技术社区 › ScalaBoy

如何选择numpy数组中行的索引?

numpy python

ScalaBoy · 技术社区 · 6 年前

我有下面的numpy数组 y_train :

我需要随机选择 n (n=2) 指数行数如下:

n=2 
n indices of rows where y=0 
n indices of rows where y=1 
n indices of rows where y=2

我使用以下代码:

n=2
idx = [y_train[np.random.choice(np.where(y_train==np.unique(y_train)[I])[0],n)].index.tolist() \
 for i in np.unique(y_train).astype(int)]

实际数组中的错误 y-列车 :

KeyError: '[70798 63260 64755 ...  7012 65605 45218] not in index'

0 回复 | 直到 6 年前

panktijk 6 年前

如果您的预期输出是中每个唯一值的随机选择索引列表 y_train :

idx = [np.random.choice(np.where(y_train == i)[0], size=2, \
       replace=False) for i in np.unique(y_train)]

输出:

[array([7, 8]), array([5, 4]), array([1, 0])]

如果要将数组展平为单个数组:

idx = np.array(idx).flatten()

输出:

array([7, 8, 5, 4, 1, 0])

Sheldore 6 年前

获得所需索引的另一种解决方案是使用 nonzero 简单的循环 range(n+1)

y_train = np.array([2,2,1,0,1,1,2,0,0])

indices = [np.random.choice((y_train==i).nonzero()[0], 2, replace=False) for i in range(n+1)]
print (indices)
# [array([7, 3]), array([5, 4]), array([0, 1])]

print (np.array(indices).ravel())
# [7 3 5 4 0 1]

推荐文章

serlingpa · 如何准备我的数据以避免无法推断频率

1 年前

Guillaume · 使用操作从Python列表创建numpy数组

2 年前

user19657580 · 在Python中打印两个numpy数组的列表

2 年前

user19657580 · Python中数组中具有相同元素的索引求和

2 年前

mikanim · 改进二维余弦函数的numpy功能

2 年前

Klimt865 · 在Python中将数组列表转换为列表列表

2 年前

theduker · 计算平均绝对误差时,If语句中赋值前引用的局部变量

2 年前

Lynn · 如果列包含Python中的特定字符串,则从列中删除值

2 年前

JasonX · 运行减法计算

2 年前

Jan Hrubec · 选择numpy数组的前n个元素

2 年前