python - 打印带有索引的 numpy 数组

Question

以下代码将打印 numpy 数组中的值，前面是数组索引。

import numpy as np
a = np.np.arange(6).reshape(2,3)
for index, val in np.ndenumerate(a):
    print(index, val)

它将打印以下内容:

(0,0) 0
(0,1) 1
(0,2) 2
(1,0) 3
(1,1) 4
(1,2) 5

有没有办法提取索引值，以便每个值都可以用类似于这样的逗号分隔打印？

0,0,0
0,1,1
0,2,2
1,0,3
1,1,4
1,2,5

Answer 1

要访问元组索引中的值，请使用它们的索引。您可以使用字符串格式来按照您想要的方式打印字符串。请参阅此了解更多信息: https://pyformat.info/

你可以这样打印:

>>> for index, val in np.ndenumerate(a):
...     print '{}, {}, {}'.format(index[0], index[1], val)
...
0, 0, 0
0, 1, 1
0, 2, 2
1, 0, 3
1, 1, 4
1, 2, 5