用python写一个删除ndarray中第五列的程序,在ndarray为空时也可用
时间: 2023-06-06 12:06:49 浏览: 36
下面是用 Python 删除 ndarray 中第五列的程序:
```python
import numpy as np
def delete_column(arr):
if arr.size == 0: # 判断 ndarray 是否为空
return np.array([]) # 若为空则返回空的 ndarray
else:
return np.delete(arr, 4, 1) # 删除第五列并返回 ndarray
# 测试样例
arr1 = np.array([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [11, 12, 13, 14, 15]])
print(delete_column(arr1)) # 输出删除第五列后的 ndarray
arr2 = np.array([]) # 空的 ndarray
print(delete_column(arr2)) # 输出空的 ndarray
```
如果您有其他问题,请继续提问。