numpy 二维数组取某一个变一维数组
时间: 2023-10-24 12:08:38 浏览: 134
可以使用 numpy 库中的 `flatten()` 方法将二维数组变成一维数组。例如:
```python
import numpy as np
# 创建一个 2x3 的二维数组
arr = np.array([[1, 2, 3], [4, 5, 6]])
# 将二维数组变成一维数组
flat_arr = arr.flatten()
print(flat_arr)
# 输出:[1 2 3 4 5 6]
```
`flatten()` 方法会将二维数组按行展开成一个一维数组。如果需要按列展开,可以使用 `flatten('F')` 方法。
相关问题
删除numpy二维数组中某行元素
要删除numpy二维数组中的某行元素,你可以使用`numpy.delete()`函数。下面是一个示例代码:
```python
import numpy as np
# 创建一个二维数组
arr = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
# 删除第二行
new_arr = np.delete(arr, 1, axis=0)
print(new_arr)
```
输出结果为:
```
[[1 2 3]
[7 8 9]]
```
在上面的示例中,我们使用`np.delete(arr, 1, axis=0)`删除了数组`arr`的第二行。`axis=0`参数表示按行操作。通过指定要删除的行索引(在这里是1),我们得到了删除指定行后的新数组`new_arr`。
NumPy一维数组索引: 一维数组切片: 二维数组索引: 二维数组切片(取第 2 列):二维数组切片(取第 2,3 行):
一维数组索引:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
print(arr[0]) # 输出:1
print(arr[2]) # 输出:3
```
一维数组切片:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
print(arr[1:4]) # 输出:[2 3 4]
```
二维数组索引:
```python
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(arr[0, 0]) # 输出:1
print(arr[1, 2]) # 输出:6
```
二维数组切片(取第 2 列):
```python
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(arr[:, 1]) # 输出:[2 5 8]
```
二维数组切片(取第 2,3 行):
```python
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(arr[1:3, :]) # 输出:[[4 5 6] [7 8 9]]
```
阅读全文