python-numpy数组如何使用元组作为索引

arr = np.zeros((2, 2), dtype=np.int32)
index = [1, 1]
# 将list转化为tuple再作为数组索引
arr[tuple(index)] = 1
print(arr)  # [[0 0],[0 1]]

你可能感兴趣的:(Python)