如何将numpy数组的后续行与第一行对齐

Question

我将打印带有以下代码的结果。

import numpy as np
a = np.array([[1,2,3],[4,5,6],[7,8,9]])
print("a[[0,1,2]]:\n","\t",a[[0,1,2]])

但是只有第一行是缩进的，如以下结果所示：

a[[0,1,2]]:
         [[1 2 3]
 [4 5 6]
 [7 8 9]]

我希望输出显示如下：

a[[0,1,2]]:
          [[1 2 3]
           [4 5 6]
           [7 8 9]]

我可以使用以下代码来实现效果：

print("a[[0,1,2]]:")
count = 0
indentation = "\t "
for raw in a[[0,1,2]]:
    count += 1
    if count == 1:
        print(indentation,"[", end = "")
        print(raw)
    elif count <= len(a[[0,1,2]])-1:
        print(indentation,"",raw)
    elif count == len(a[[0,1,2]]):
        print(indentation,"",raw, end="")
        print("]")

但是我认为这不是显示显示的方便方法结果。有什么方法可以将整个数组直接移动到输出窗口中的右侧？

Answer 1

这是一个更简单，通用，填充功能：

def pad(a, sep='\t'):
    from itertools import repeat
    t = repeat(sep)
    return '\n'.join(map(''.join, zip(t, str(a).split('\n'))))

print("a[[0,1,2]]:\n", pad(a[[0,1,2]]))

输出：

a[[0,1,2]]:
    [[1 2 3]
     [4 5 6]
     [7 8 9]]