0 python numpy multidimensional-array numpy-ndarray
我是 numpy 新手,我正在尝试创建添加新行。我读到我只需要指定轴。
现在我只是在测试
import numpy as np
arr = np.array(['1', '2', '3'])
print(np.append(arr, np.array(['4', '5', '6']), axis = 0))
Run Code Online (Sandbox Code Playgroud)
但这总是输出 ['1' '2' '3' '4' '5' '6']。我哪里出错了?
您的数组只有一个维度(轴),因此您无法将新数据添加到另一个维度。你首先需要重塑:
out = np.append(arr[None], np.array([['4', '5', '6']]), axis=0)
Run Code Online (Sandbox Code Playgroud)
或者:
out = np.append(arr.reshape(1, -1), np.array([['4', '5', '6']]), axis = 0)
Run Code Online (Sandbox Code Playgroud)
也就是说,vstack可能更适合您的用例:
out = np.vstack([arr, np.array(['4', '5', '6'])])
Run Code Online (Sandbox Code Playgroud)
输出:
array([['1', '2', '3'],
['4', '5', '6']], dtype='<U1')
Run Code Online (Sandbox Code Playgroud)