Python – Numpy 数组列删除

在使用Python的NumPy模块时，我们通常需要删除数组中的某些列。本篇文章将讨论如何使用NumPy删除数组的列。

Numpy数组

NumPy是Python科学计算中最受欢迎的库之一，用于对多维数组和矩阵进行处理。NumPy提供了一个非常有用的对象，即ndarray对象（N维数组对象），它是一种多维数组对象，用于存储同类型的数据。

删除数组列

要删除NumPy数组中的列，需要使用np.delete()函数。该函数接受三个参数，第一个参数是要处理的数组，第二个参数是要删除的列号，第三个参数是轴。

以下是一个简单的示例，展示了如何使用np.delete()函数删除NumPy数组中的列：

import numpy as np
 
#创建一个3x4的数组
arr = np.array([[1,2,3,4], [5,6,7,8], [9,10,11,12]])
print("原始数组：")
print(arr)
 
# 删除第二列
newarr = np.delete(arr, 1, 1)
print("\n删除第二列后的新数组：")
print(newarr)

上面的代码将输出以下结果：

原始数组：
[[ 1  2  3  4]
 [ 5  6  7  8]
 [ 9 10 11 12]]

删除第二列后的新数组：
[[ 1  3  4]
 [ 5  7  8]
 [ 9 11 12]]

在上面的示例中，我们创建了一个3x4的NumPy数组，然后将第二列从数组中删除。删除后，我们使用print()函数输出新数组的内容。

轴参数

np.delete()函数还包括一个可选的轴参数，用于指定要删除的行或列。默认情况下，该参数的值为None，表示删除数组的一个扁平化版本。如果该参数的值为0，则表示删除行，为1则表示删除列。

以下是一个示例，展示了如何在NumPy数组中删除行：

import numpy as np
 
#创建一个3x4的数组
arr = np.array([[1,2,3,4], [5,6,7,8], [9,10,11,12]])
print("原始数组：")
print(arr)
 
# 删除第二行
newarr = np.delete(arr, 1, 0)
print("\n删除第二行后的新数组：")
print(newarr)

上面的代码将输出以下结果：

原始数组：
[[ 1  2  3  4]
 [ 5  6  7  8]
 [ 9 10 11 12]]

删除第二行后的新数组：
[[ 1  2  3  4]
 [ 9 10 11 12]]

在上面的示例中，我们使用np.delete()函数删除了NumPy数组中的第二行，并输出了新数组的内容。

总结

在Python的NumPy模块中，我们可以使用np.delete()函数删除NumPy数组中的列或行。该函数接受三个参数，第一个参数是要处理的数组，第二个参数是要删除的列号或行号，第三个参数是轴。删除后，我们可以使用print()函数输出新数组的内容。

欢迎阅读本篇文章，祝学习愉快！