python - 子集和转置python数组

Question

我有一个数组：

[['a','1'.'2','3','4'],['b','5','6','7','8'],['c','9','10','11','12'],['d','13','14','15','16']]

我想对此进行子集化并转置。对于转置，我尝试了 zip 函数，但它输出的数组在我写入 csv 文件时没有制表符分隔的格式。

所需的输出：

a   b
1   5
2   6
3   7
4   8
c   d
9   13
10   14
11   15
12   16

由于我无法在此处制作制表符，因此我在输出中留下了三个空格。

Answer 1

这给出了看起来像您正在寻找的输出：

import numpy as np

a = np.array([['a','1','2','3','4'],
              ['b','5','6','7','8'],
              ['c','9','10','11','12'],
              ['d','13','14','15','16']])

new_a = np.hstack((a[:2,], a[2:])).T

结果：

>>> new_a
array([['a', 'b'],
       ['1', '5'],
       ['2', '6'],
       ['3', '7'],
       ['4', '8'],
       ['c', 'd'],
       ['9', '13'],
       ['10', '14'],
       ['11', '15'],
       ['12', '16']], 
      dtype='|S2')