使用标记

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

要访问元组index中的值，请使用它们的索引。您可以使用字符串格式化打印字符串的方式。有关更多信息，请参阅此 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

Answer 2

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

这对我有用。如果数组会改变大小，你可能想要动态地进行打印部分，但如果它是一段代码，我认为这很好。