import numpy as np
a=np.array([1,1,2,3,3,4,4])
b=np.unique(a,return_counts=True)
print(b[0][list(b[1]).index(1)])
想问一下代码中print()里面的是怎么计算的。
b[0][list(b[1]).index(1)]结果为什么是2,运算步骤是啥。
import numpy as np
a=np.array([1,1,2,3,3,4,4])
b=np.unique(a,return_counts=True)
print(b[0][list(b[1]).index(1)])
想问一下代码中print()里面的是怎么计算的。
b[0][list(b[1]).index(1)]结果为什么是2,运算步骤是啥。
这段代码是用来找到a中最小的无重复元素。
>>> import numpy as np
>>> a=np.array([1,1,5,2,3,3,4,4])
>>> b=np.unique(a,return_counts=True)
>>> b
(array([1, 2, 3, 4, 5]), array([2, 1, 2, 2, 1], dtype=int64))
>>> type(b[0]),type(b[1])
(<class 'numpy.ndarray'>, <class 'numpy.ndarray'>)
>>> list(b[1]).index(1)
1
>>> b[0][list(b[1]).index(1)]
2
>>>
首先,np.unique用来去掉a中所有的重复元素,并从小到大排序,也就是b[0]。同时return_counts参数设为True,则会统计b[0]中每个元素重复了几次。
然后,list(b[1]).index(1)把b[1]转换成list列表,用list的index方法找出b[1]中第一个值为1的元素的下标1,这里值1表示没有重复,下标1对应b[0][1],也就是2。