【问题标题】:How to add an extra row of 1's at top of a NumPy array [duplicate]如何在 NumPy 数组顶部添加额外的 1 行 [重复]
【发布时间】:2019-06-25 01:35:54
【问题描述】:

数组显示为:

a[[12,31,5], [5,32,1]]

我希望添加一行 1,使其变为:

a[[1,1,1], [12,31,5], [5,32,1]]

怎么做?

【问题讨论】:

标签: python-3.x numpy


【解决方案1】:

你只需要np.vstack

a= np.array([[12,31,5], [5,32,1]])
np.vstack([np.ones(a.shape[1]),a])

array([[ 1.,  1.,  1.],
       [12., 31.,  5.],
       [ 5., 32.,  1.]])

稍微复杂一点的是np.r_

np.r_[np.ones(a.shape[1]).reshape((1,-1)),a]
array([[ 1.,  1.,  1.],
       [12., 31.,  5.],
       [ 5., 32.,  1.]])

【讨论】:

    猜你喜欢
    • 2012-01-19
    • 2011-03-02
    • 1970-01-01
    • 1970-01-01
    • 2019-05-15
    • 1970-01-01
    • 2018-10-28
    • 1970-01-01
    相关资源
    最近更新 更多