仅特定列中的唯一值二维数组(使用numpy)

时间:2019-07-29 14:30:35

标签: arrays numpy numpy-ndarray

我在2-d中有一个numpy数组。我希望仅在特定列中获取唯一值。

        import numpy as np
        data = np.genfromtxt('somecsvfile',dtype='str',delimiter=',')

        #data looks like
        [a,b,c,d,e,f,g],
        [e,f,z,u,e,n,c],
        ...
        [g,f,z,u,a,v,b]


仅使用numpy/scipy,如何获取第5列中的数组或唯一值列表。 (我知道用熊猫很容易做到。)

预期的输出为2个值:[e,a]

1 个答案:

答案 0 :(得分:0)

正确答案已发布。本质上是一个简单的引用问题。

np.unique(数据[:,4])

谢谢。