python删除数组第i列
时间: 2023-11-17 10:04:15 浏览: 86
数组数据删除
要删除Python中数组的第i列,你可以使用列表推导式或NumPy库来实现。这里提供两种方法:
1. 使用列表推导式:
假设你有一个二维数组`my_array`,可以通过以下方式删除第i列:
```python
my_array = [[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]
i = 1 # 要删除的列索引(从0开始计数)
new_array = [row[:i] + row[i+1:] for row in my_array]
print(new_array)
```
输出结果:
```
[[1, 3],
[4, 6],
[7, 9]]
```
2. 使用NumPy库:
如果你使用NumPy库处理数组,可以使用`delete`函数删除指定的列:
```python
import numpy as np
my_array = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
i = 1 # 要删除的列索引(从0开始计数)
new_array = np.delete(my_array, i, axis=1)
print(new_array)
```
输出结果:
```
[[1 3]
[4 6]
[7 9]]
```
以上两种方法都能够删除数组中的指定列。选择哪种方法取决于你使用的数据结构和库的需求。
阅读全文