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

2023年6月15日 205次阅读

我试图使用
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只狐狸,也缺少一些独特的条目.当然这只是虚拟数据,但我希望我能清楚地解释发生了什么.

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

Gridiron
Cricket
Ping Pong Table
Softball

UPDATE2：

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

谢谢

最佳答案
str.lstrip(' ')不是就地方法,它返回剥离的字符串.你需要将它分配回对象 –

object = object.lstrip(' ')

假设Python 2.7(或3.1),更快的方法是使用set,也许设置理解.示例 –

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

请注意,这不会保留任何订单,因为订单不是订购的.如果订单很重要,您可以使用一个集来存储已经看到的值.示例 –

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(”).具有集合理解的条带示例 –

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