[英]Insert 1D NumPy array as column in existing 2D array
I have a 2D
NumPy
array: 我有一个
2D
NumPy
数组:
>>> import numpy as np
>>> a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
>>> a
array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
and a 1D
array: 和一
1D
数组:
>>> b = np.arange(3)
>>> b
array([0, 1, 2])
Is there an elegant way to insert b
into a
as a new first column? 有没有一种优雅的方法可以将
b
插入a
作为新的第一列?
So that: 以便:
>>> a
array([[0, 1, 2, 3],
[1, 4, 5, 6],
[2, 7, 8, 9]])
You could use column_stack()
您可以使用
column_stack()
In [256]: np.column_stack((b, a))
Out[256]:
array([[0, 1, 2, 3],
[1, 4, 5, 6],
[2, 7, 8, 9]])
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.