NumPy中insert()与concatenate()

2018-11-15  本文已影响10人  juriau

numpy.concatenate((a1, a2, ...), axis=0)
放到第axis维的最后。

>>> a = np.array([[1, 2], [3, 4]])
>>> b = np.array([[5, 6]])
>>> np.concatenate((a, b), axis=0)
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> np.concatenate((a, b.T), axis=1)
array([[1, 2, 5],
       [3, 4, 6]])

numpy.insert(arr, obj, values, axis=None)
在arr每个第obj个放values,在第axis维上。(you know歪果仁都喜欢最后写in xxx)

>>> a = np.array([[1, 1], [2, 2], [3, 3]])
>>> a
array([[1, 1],
       [2, 2],
       [3, 3]])
>>> np.insert(a, 1, 5)
array([1, 5, 1, 2, 2, 3, 3])
>>> np.insert(a, 1, 5, axis=1)
array([[1, 5, 1],
       [2, 5, 2],
       [3, 5, 3]])
上一篇下一篇

猜你喜欢

热点阅读