Question

我想在X_test和y_test中删除MFD较大的每一行。问题是，我总是得到Train / Test / Split的随机混合索引。如果我尝试删除它，我会收到以下错误消息：

IndexError：索引3779超出了轴1的大小为3488

的范围

我不能使用旧指数来放弃它，但我怎样才能获得MFD＆gt;的新指数。 1

  X_train, X_test, y_train, y_test = train_test_split(X, y, 
                                                test_size=test_size, 
                                                random_state=random_state, 
                                                stratify=y)



mfd_drop_rows = []
i_nr = 0
for i in X_test.MFD:
   if (i > 1): 
      mfd_drop_rows.append(X_test.index[i_nr])
   i_nr += 1


X_test_new = X_test.drop(X_test.index[mfd_drop_rows]) 
y_test_new = Y_test.drop(Y_test.index[mfd_drop_rows])

感谢您的帮助（=

Answer 1

我解决了它，我只是使用我的i_nr迭代并拥有新的指标。

感谢所有阅读它的人

mfd_drop_rows = []
i_nr = 0
for i in X_test.MFD:
 if (i > 1): 
    mfd_drop_rows.append(i_nr)
 i_nr += 1


 X_test_new = X_test.drop(X_test.index[mfd_drop_rows]) 
 y_test_new = Y_test.drop(Y_test.index[mfd_drop_rows])

Answer 2

不确定MFD是什么，但假设X_test.MFD为您提供了一组数字，您可以使用掩码来删除行。这里可以看到一个如何使用面具的简单示例：

x = [[1,2,3,4,5],[6,7,8,9,10]]
mfd = [0.6, 1.3]
mask = x > 1
x_new = x[mask,:]

这会给：

x = [1,2,3,4,5
     6,7,8,9,10]
mask = [False, True]
x_new = [6,7,8,9,10]

Python /如何在/ train / test / split之后使用索引删除测试数据中的特定行

2 个答案: