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。
np.array([1,1,2,3,3,4,4])创建一个数组:array([1, 1, 2, 3, 3, 4, 4])
np.unique() 函数 去除其中重复的元素 ,并按元素 由小到大 返回一个新的无元素重复的元组或者列表。
加上参数np.unique(a,return_counts=True),返回两个列表,一个是np.unique() 函数普通返回的元素,另一个则是对应元素在旧列表里各自出现了几次
https://blog.csdn.net/weixin_44211968/article/details/123772201