数组增加一列
时间: 2023-09-05 20:07:09 浏览: 86
一个递增的数列
增加一列可以使用numpy或pandas库中的函数实现。以下是两个简单的示例代码:
使用numpy库:
```python
import numpy as np
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
new_col = np.array([10, 11, 12])
data_with_new_col = np.column_stack((data, new_col))
print(data_with_new_col)
```
这个例子中,我们使用numpy库定义一个包含多行数据的数组`data`,然后定义一个新的一维数组`new_col`,最后使用`column_stack`函数将两个数组按列合并。结果为一个包含四列的二维数组。
使用pandas库:
```python
import pandas as pd
import numpy as np
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
df = pd.DataFrame(data, columns=['col1', 'col2', 'col3'])
new_col = pd.Series([10, 11, 12], name='col4')
df_with_new_col = pd.concat([df, new_col], axis=1)
print(df_with_new_col)
```
这个例子中,我们使用numpy库定义一个包含多行数据的数组`data`,然后使用pandas库将数组转换为DataFrame对象,并指定DataFrame的列名。接着,我们定义一个新的Series对象`new_col`,并指定Series的名称。最后,我们使用`concat`函数将原DataFrame对象和新Series对象按列合并。结果为一个包含四列的DataFrame对象。
阅读全文