ndarray删去一列
时间: 2023-02-17 13:03:57 浏览: 95
numpy.ndarray 实现对特定行或列取值
可以使用 numpy 库中的 delete() 函数来删除 ndarray 中的一列。使用方法如下:
```
import numpy as np
# 初始化ndarray
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
# 删除第二列
arr = np.delete(arr, 1, axis=1)
print(arr)
```
输出结果为:
```
[[1 3]
[4 6]
[7 9]]
```
参数说明:
- `arr`: 要删除一列的ndarray
- `1`: 要删除的列的索引
- `axis=1`: 指定删除的维度为列(如果是行,则指定 axis=0)
如果你要删除多列,可以把要删除的列索引放在一个数组里,例如:
```
arr = np.delete(arr, [1,2], axis=1)
```
这样就可以删除第2,3列了。
阅读全文