如何使用Python检查CSV文件中是否已存在数据

Question

大家好

我正在尝试编写一个从文本文件中获取IP地址的python脚本，检查CSV文件中是否已存在IP地址，如果已经存在IP地址，则不执行任何操作（通过），如果IP不存在在CSV文件中添加IP地址，然后在CSV文件中添加IP地址。

我可以将IP地址添加到CSV文件中，但是我无法理解如何检查文件中是否已存在IP地址。

if os.path.isfile('IP_file.txt'):
    logfile = open('IP_file.txt', 'r')
    csv_file = open('csv_file.csv', 'ab')
    for line in logfile:
        if line in csv_file: # This is not working
            pass
        else:
            csv_file.write(line)
    csv_file.close()
    logfile.close()
else:
    print('No file with filename ' logfile 'created')

我也尝试使用CSV模块，但没有运气。 任何帮助深表感谢。

Answer 1

CSV只是一种文本格式。 逐行阅读，针对您的IP测试每一行。 喜欢

ip="192.168.1.1"
for line in csv_file:
    if ip in line:
        found = True
        break

我有点担心您对磁盘上的文件进行循环。 最好将文件读入内存（首先在CSV文件中找到所有IP，然后放入列表中），然后对照该列表进行检查，而不是为IP文件的每一行打开并遍历整个CSV文件。

Answer 2

保留一个包含唯一获取的值的列表或字典：

found_address = []
with open('IP_file.txt', 'r') as logfile:
    with open('csv_file.csv', 'ab') as csv_file:
        for line in logfile:
            if not line in found_address:
                csv_file.write(line)
                found_address.append(line)

Answer 3

您正在以追加模式ab打开csv_file.csv。 然后，您尝试检查line in csv_file 。 在追加模式下，文件指针指向文件结尾。

Answer 4

小小的疯狂的函数式python：

ip_file = set(map(str.strip, open('IP_file.txt').readlines()))

如果CSV文件是： 192.168.1.1,192.168.1.2,192.168.1.3，...

csv_file = set(map(str.strip, open('csv_file.csv').read().split(',')))
diff_ip = ip_file - csv_file
open('csv_file.csv','a').write(''.join(map(lambda x: ',{}'.format(x), list(diff_ip))))