python – 将平面数组值转换为列向量

2023年3月1日 526次阅读

我有一个表示为字符串的基数为3的数组：

['1', '2', '10']

我想对每个数字进行0填充,使得每个数字占用的最大空间为3.

['001', '002', '010']

然后将其转换为以下矩阵：

[[0, 0, 0],
[0, 0, 1],
[1, 2, 0]]

也就是说,将每个字符串条目转换为列向量.我尝试过旋转,转置,并且不确定最好的方法是什么.

谢谢

最佳答案 使用str.zfill填充零,然后使用np.dstack转换预期的格式：

In [106]: np.dstack([list(i.zfill(3)) for i in a])[0].astype(np.int)
Out[106]: 
array([[0, 0, 0],
       [0, 0, 1],
       [1, 2, 0]])
点赞