python - 解析二维数组中的字符串推送值

Question

我想将一串数字解析为二维方阵/矩阵，例如第一个数字标识矩阵的大小

输入45621797533863034这里第一个数字 4 标识其 4x4 矩阵，其余是值。

所以数组应该是

5 6 2 1
7 9 7 5
3 3 8 6
3 0 3 4

Answer 1

IIUC:

s = '45621797533863034'
s = np.array(list(s))
s[1:].reshape(int(s[0]), -1)

输出:

array([['5', '6', '2', '1'],
       ['7', '9', '7', '5'],
       ['3', '3', '8', '6'],
       ['3', '0', '3', '4']], dtype='<U1')

如果你希望输出为数字，你可以传入正确的数据类型:

s = '45621797533863034'
s = np.array(list(s), dtype=np.uint8)
s[1:].reshape(s[0],-1)

输出:

array([[5, 6, 2, 1],
       [7, 9, 7, 5],
       [3, 3, 8, 6],
       [3, 0, 3, 4]], dtype=uint8)