Python删除首元素与前一行重复的行问题求助

先说说你代码的核心问题：

• for i in data是遍历数组里的每一行元素，不是索引。比如第一次循环i是[0.000, 1]，你用data[i, 0]相当于把数组当索引用，numpy自然会报类型错误——因为索引必须是整数或布尔值。
• 就算改成索引循环，np.delete也不会直接修改原数组，它会返回删除后的新数组，不把结果赋值回去等于白操作；而且正向循环删元素会导致后续索引错位，很容易删错行。

给你两种靠谱的解决方法：

方法一：numpy差异计算+布尔掩码（推荐） #

用numpy的向量操作替代循环，高效且不易出错：

import numpy as np

# 先把数据转成标准numpy数组
data = np.array([[0.000, 1], [0.0025, 2], [0.0025, 3], [0.005, 5]])

# 计算首列相邻元素的差值
diff_first_col = np.diff(data[:, 0])
# 生成掩码：第一行必须保留，后续行仅当与前一行首列不同时保留
mask = np.concatenate([[True], diff_first_col != 0])
# 过滤得到去重后的数组
filtered_data = data[mask]

print(filtered_data)
# 输出：[[0.    1.   ]
#        [0.0025 2.   ]
#        [0.005  5.   ]]

方法二：反向遍历索引删除 #

如果非要用循环，必须从后往前遍历索引，避免删除操作导致的索引错位：

import numpy as np

data = np.array([[0.000, 1], [0.0025, 2], [0.0025, 3], [0.005, 5]])

# 从倒数第二行开始向前遍历
for i in range(len(data)-2, -1, -1):
    if data[i, 0] == data[i+1, 0]:
        # 将删除后的新数组重新赋值给data
        data = np.delete(data, i+1, axis=0)

print(data)
# 输出与方法一一致

反向遍历的原因：如果正向删除，删完某一行后，后续行的索引会前移，导致循环判断时跳过部分元素；反向遍历则不会影响前面元素的位置，能准确删除目标行。

内容的提问来源于stack exchange，提问作者vinc00