Python中唯一的对象列表无法正常工作

Question

我试图使用Python创建一个唯一的对象列表，我失败了。 无论我使用列表还是集合都没关系，它似乎不起作用。 当我打印列表/集时，我注意到列表中有几个非唯一对象。 我意识到情况就是这样，因为其中一些对象在单词的开头有一个“空格”。 我环顾四周，想到使用·lstrip（''）将有助于我的事业，但遗憾的是它没有。

最奇怪的是“唯一对象的数量”是正确的，但最后创建的唯一对象列表是错误的。 任何人都可以指出我哪里出错了？

我感兴趣的专栏是'对象'，唯一的列表应该包含猫头鹰 ， 猫 ， 狐狸 ， 牛 ， 山羊 ， 狗 ， 蚂蚁 ， 水牛 ， 狮子和老虎 。

样本数据：

Key    ID    Name    Code    State    Object
01     NULL  NULL   NULL    NULL      Athletics, Light,Netball
02     NULL  NULL   NULL    NULL      BMX Track, Gridiron, Oval
05     NULL  NULL   NULL    NULL      Dog park, Cricket, Soccer
10     NULL  NULL   NULL    NULL      Netball, Oval, Softball
21     NULL  NULL   NULL    NULL      Seat, Playground, Ping Pong Table
13     NULL  NULL   NULL    NULL      Bench, Bike Rack, Seat

我的工作代码如下：

import csv

fOpen1=open('C:\Data.csv')
uniqueList=csv.writer(open('C:\UniqueList.csv', 'wb'))

Master=csv.reader(fOpen1)
Master.next()

unique=[]

for row in Master:
    for item in row[5].split(','):
        item.strip(' ')
        if item not in unique:
            unique.append(item)
uniqueList.writerow(unique)

我最后得到的是重复，包括2只狐狸，也缺少一些独特的条目。 当然这只是虚拟数据，但我希望我能清楚地解释发生了什么。

更新1：我已经更新了脚本，但它可以正常工作但是又出现了另一个问题。 我已经使用我正在使用的真实数据更新了该列。 未添加到最终列表的唯一项包括：

Gridiron
Cricket
Ping Pong Table
Softball

UPDATE2：

我已经恢复到原来的'错误'脚本，因为它现在可以正常工作了。 我正在处理的csv文件有问题。

谢谢

Answer 1

str.lstrip(' ')不是就地方法，它返回剥离的字符串。 你需要将它分配回object -

object = object.lstrip(' ')

假设Python 2.7+（或3.1+），更快的方法是使用set ，也许set comprehension 。 示例 -

unique = {obj.lstrip() for row in Master for obj in row[5].split(',')}
uniqueList.writerow(list(unique))

请注意，这将不保存任何顺序，因为set s的不排序。 如果订单很重要，您可以使用一个set来存储已经看到的值。 示例 -

unique=[]
seen_set = set()
for row in Master:
    for obj in row[5].split(','):
        obj = obj.lstrip(' ')
        if obj not in seen_set:
            unique.append(obj)
            seen_set.add(obj)

另外，我想建议您不要将object用作变量名，因为它是内置类的名称（由所有其他类扩展）。

---

此外，似乎最后有一些带有空格的字符串，因此最好使用.strip()或.strip(' ')而不是.lstrip(' ') 。 具有集合理解的strip带示例 -

unique = {obj.strip() for row in Master for obj in row[5].split(',')}
uniqueList.writerow(list(unique))

Answer 2

编辑你的代码是这样的：

for object in row[5].split(','):
        object=object.strip()
        if object not in unique:
            unique.append(object)

strip将从右侧和左侧删除所有空格。并将对象分配为新对象as

object = object .strip()  

Answer 3

一套理解能为你服务。

首先，让我们使用上下文管理器删除该打开的文件：

import csv

with open('C:\Data.csv') as raw:
    master = csv.reader(raw)
    master.next()  # Ignore the header
    unique = {y.strip() for row in master for y in row[-1].split(',')}

好吧，让我们回顾一下我们在那里做的事情。 我们使用上下文管理器打开文件，因此文件将自动关闭。 然后我们使用csv.reader读取csv并迭代超过第一行。

这就是它变得棘手的地方 - 我们通过迭代csv中的列表创建了一个集合，然后迭代这些列表的内容。 一个更冗长的方式：

unique = set()
for row in master:
    for item in row[-1].split(','):
        unique.add(item.strip())

这实现了很多相同的事情，可能是一种更容易理解的格式。 另外，请注意我使用-1切片到csv中的最后一列。