[英]numpy array to list formatting
我如何格式化一個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]]]
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.