在if语句中使用For循环

Question

我有一个380行20列的表格。我想从该表中删除符合特定条件的行。

为了澄清问题，假设我有以下列表：

names = ['John', 'Amy', 'Daniel']

我想删除在列表名称中找到姓名的所有人员的数据。

例如，假设我的数据如下所示：

John    82    3.12    boy
Katy    12    1.12    girl
Amy     42    2.45    girl
Robert  32    1.56    boy
Daniel  47    2.10    boy

我想删除John、Amy和Daniel的数据。所以输出应该是：

Katy    12    1.12    girl
Robert  32    1.56    boy

import csv
import numpy as np

# loading data
data = np.genfromtxt('file.txt', dtype = None)

csvfile = "home/paula/Desktop/test.txt"
with open(csvfile, 'w') as output:
    writer = csv.writer(output, delimiter = '\t')

    for row in range(len(data)):
        if data[row][0] == (i for i in names):
            print 'removing the data of', i, '...'
        else:
            writer.writerow([data[row][0], data[row][1], 
                             data[row][2], data[row][3]])

我的代码正在运行，但数据没有从我的原始数据中删除。当我打开新的测试.txt文件时，我可以看到数据没有被删除。

我可以肯定错误在< code > if data[row][0]= =(I for I in names):我该如何修复这个问题？

Answer 1

if data[row][0] == (i for i in names):
            print 'removing the data of', i, '...'

在该部分中，i在（i for i in name）中用作本地可验证。但在下一个打印行中，您使用i。在这里您不能使用这个。

您可以使用检查，就好像名称中的数据[行][0]一样：您可以尝试如下操作：

if data[row][0] ==  names:
            print 'removing the data of', data[row][0], '...'

Answer 2

您正在检查数据[row][0]是否与（i表示名称中的i）。您要做的是检查它是否与＜code＞（名称中的i表示i）

any([data[row][0]==i for i in names])

您也可以使用运算符中的＜code＞以非荒谬的方式执行：

data[row][0] in names

这将检查name的任何元素是否与data[row][0]相同。

Answer 3

条件应写为：

if data[row][0] in names:

在您当前的代码中，＜code＞（i代表名称中的i）

>>> (i for i in names)
<generator object <genexpr> at 0x1060564b0>
>>> 'John' == (i for i in names)
False
>>>

相反，您可以测试项目是否在列表中，如下所示

>>> names = ['John', 'Amy', 'Daniel']
>>> 'John' in names
True
>>> 'Bob' in names
False
>>>

如注释中所述，通过在迭代行之前将名称转换为集合，可以使此检查更加高效。但理想情况下，您将使用熊猫库来操作csv /表数据。有关类似示例，请参阅此答案。您可以使用 df[~df.名称.