python - 显示数组中某一列中特定数字的所有值

Question

对于第一列中的特定值，如何打印数组中的所有其他数字？

示例:假设我有 4 列，如下所示:

[1   1   1]
[1   1   2]
[1   2   1]
[1   2   2]
[2   1   1]
[2   1   2]
[2   2   1]
[2   2   2]

对于第一列中的每个唯一数字，是否有一种直接的方法可以提取第二列和第三列的数字？

我知道我可以按列对数组进行排序，找到每列中的唯一值，并找到这些唯一值的索引。我想我想要的是:

col1val = 1
col2vals = [[1], [1], [2], [2]]
col3vals = [[1], [2], [1], [2]]

(或类似的东西)。

编辑: 非常感谢 @Randrian 的建议，让我做到了这一点:

unique, indices = np.unique(a[:,0], return_index=True)
print(a[indices[0]:indices[1],1])  # column 2

给出:

[1 1 2 2]

Answer 1

如果你使用 numpy，你可以使用:

a = np.array([[1   1   1],
[1   1   2],
[1   2   1],
[1   2   2],
[2   1   1],
[2   1   2],
[2   2   1],
[2   2   2]])

并通过以下方式提取唯一值:

indices = np.unique(a[:,0])

我不清楚第一列中的唯一数字到底想要做什么。您可以将它们用作第二列和第三列的索引:

print(a[indices, 1])  # column 2
print(a[indices, 2])  # column 3