要从NumPy结构化数组中删除一列,可以使用numpy.delete()
函数。该函数可以删除指定列,并返回新的数组。
以下是使用NumPy删除结构化数组中一列的步骤:
import numpy as np
arr = np.array([(1, 2, 3), (4, 5, 6)], dtype=[('col1', int), ('col2', int), ('col3', int)])
new_arr = np.delete(arr, 'col2', axis=1)
在上述代码中,我们首先导入了NumPy库,并使用np.array()
函数创建了一个结构化数组。结构化数组中的每个元素都由3列组成,分别是'col1'、'col2'和'col3'。然后,我们使用np.delete()
函数删除了名为'col2'的列,并通过axis=1
参数指定删除的是列而不是行。最后,将新的数组赋值给new_arr
变量。
删除指定列后,new_arr
将不再包含被删除的列。此操作可以确保在将结构化数组保存为二进制文件时,被删除的列不会出现在其中。
请注意,本答案中的示例代码和解释使用了NumPy库,该库是一个流行且常用的科学计算库。
领取专属 10元无门槛券
手把手带您无忧上云