Question

我是python的新手，我有两个csv文件。第一个（more.csv）的内容为

A123,B456,C789

第二个（less.csv）的内容为

B456

我想这样，当它们相同时，它们将相似的项目存储到一个名为“same”

的变量中

我认为它会从以下内容开始：

more = open('more.csv','r')
less= open('less.csv','r')
for item in unitid:

谢谢。

Answer 1

如果它们只有一行，您可以使用 set 对象（内置python）来比较它们，例如：

>>> a = ['A123','B456','C789','D007']
>>> b = ['B456','D007','E009']
>>> c = list(set(a).intersection(b))
>>> print c
['B456', 'D007']

从文件中比较的完整方法如下所示：

def compare( fileA, fileB ):
    a_file = open(fileA, 'r')
    a_data = a_file.read()
    a_file.close()

    b_file = open(fileB, 'r')
    b_data = b_file.read()
    b_file.close()

    # compare the contents
    a_set = set(a_data.split(','))
    b_set = set(b_data.split(','))

    return list(a_set.intersection(b_set))

compare('more.csv', 'less.csv')

如果它们每个文件只有一行，那么你仍然可以使用它，你只需要修改一下 - 我想将交叉点存储到一个代表line-by的数组中 - 相似之处。

Answer 2

将CSV文件处理为列表后，您可以使用collections.Counter查找重复项：

from collections import Counter

# after processing your CSV files into two lists:
more_list = ['A123', 'B456', 'C789', 'D007']
less_list = ['B456', 'D007', 'E009']

dupe_counter = Counter(more_list)
dupe_counter.update(less_list)
same_list = [val for val in dupe_counter if dupe_counter[val] > 1]
# same_list will be: ['B456', 'D007']

比较两个csv文件内容并将类似的内容存储到变量中

2 个答案: