使用np.array()函数创建一维数组和二维数组,参数为列表
时间: 2024-09-25 10:17:44 浏览: 55
Python numpy实现二维数组和一维数组拼接的方法
在NumPy库中,`np.array()`函数常用于将列表转换成数组,无论是单一维度的一维数组还是多维度的二维数组。以下是创建这两种数组的例子:
**一维数组:**
如果你有一个包含数值的列表,可以这样创建一维数组:
```python
import numpy as np
# 列表数据
data_list = [1, 2, 3, 4, 5]
# 转换为一维数组
one_dim_array = np.array(data_list)
print(one_dim_array)
```
**二维数组:**
如果你需要创建一个矩阵,可以使用嵌套的列表表示每个元素所在的行和列:
```python
# 行列表和列列表
rows = [[1, 2], [3, 4], [5, 6]]
columns = [[7, 8], [9, 10], [11, 12]]
# 创建二维数组
two_dim_array = np.array(rows)
# 或者直接提供两个嵌套列表作为参数
two_dim_array_alternative = np.array([[7, 8], [9, 10], [11, 12]])
print(two_dim_array)
print(two_dim_array_alternative)
```
阅读全文