又一个表明list和array的不同

import numpy as np
a = np.array([1, 2, 3])
b = np.array([3, 1, 2])
c = np.array([4, 5, 6])
c[a > b] = 0
print(c)   # 【0, 5, 6】
print(a > b)  #[False, True, True]
# 这说明 c[a>b] =0 是把a>b的那些index返回了,所以会把为True的全部调整为0

# 从下面还可以看出来列表与array的区别
d = [1, 2, 3]
e = [3, 1, 2]
f = [4, 5, 6]
f[d > e] = 0
print(f)   # [0, 5, 6]
print(d > e)  # False

猜你喜欢

转载自blog.csdn.net/qfpkzheng/article/details/79693678