我将如何格式化一个 numpy 形式的数组
data1 = np.array([[0,0,0],[0,1,1],[1,0,1],[1,1,0]])
以这种格式的列表:
data = [
[[0,0], [0]],
[[0,1], [1]],
[[1,0], [1]],
[[1,1], [0]]
]
我尝试使用两个 for 循环
for i in range(len(data)):
for j in range(3):
if j == 2:
va[i] = data1[i][j]
else:
sa[i] = data1[i][j]
但这给了我一个索引越界错误。我很想知道如何解决这个问题
最佳答案
使用列表理解和切片:
>>> data1 = np.array([[0,0,0],[0,1,1],[1,0,1],[1,1,0]])
>>> print [[x[:2].tolist(), x[2:].tolist()] for x in data1]
[[[0, 0], [0]],
[[0, 1], [1]],
[[1, 0], [1]],
[[1, 1], [0]]]
关于python - numpy 数组列表格式,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/20450431/