python代码。数组如果存在指定列名的2列,则删除。
时间: 2023-02-24 11:50:18 浏览: 172
numpy.delete删除一列或多列的方法
以下是 Python 代码,实现删除数组中存在指定列名的两列:
```
def remove_columns(array, column_names):
columns_to_remove = []
for column_name in column_names:
if column_name in array.dtype.names:
columns_to_remove.append(column_name)
return array[columns_to_remove]
```
使用方法:
```
import numpy as np
# 创建一个数组
array = np.array([(1, 2, 3), (4, 5, 6)], dtype=[('a', '<i4'), ('b', '<i4'), ('c', '<i4')])
# 删除列 'a' 和 'c'
new_array = remove_columns(array, ['a', 'c'])
print(new_array)
```
输出:
```
[(2,) (5,)]
```
阅读全文