python - 将数组中的每个值复制为一个新数组

Question

我有一个像这样的 numpy ndarray 输入

[[T, T, T, F],
  [F, F, T, F]]

我想将每个值复制为一个新数组，所以输出将是

[[[T,T], [T,T], [T,T], [F,F]]
  [[F,F], [F,F], [T,T], [F,F]]]

我该怎么做？提前谢谢你

Answer 1

一种方法是使用 np.dstack沿第三轴复制数组:

np.dstack([a, a])

array([[['T', 'T'],
        ['T', 'T'],
        ['T', 'T'],
        ['F', 'F']],

       [['F', 'F'],
        ['F', 'F'],
        ['T', 'T'],
        ['F', 'F']]], dtype='<U1')

---

设置:

T = 'T'
F = 'F'
a = np.array([[T, T, T, F],
              [F, F, T, F] ])