2
svar
114
visningar
Sammanfoga csv filer
Hej!
Jag vet hur man använder intersection. Exempelvis,
>>> a = {1, 2, 4, 7}
>>> b = {3, 4, 9, 11}
>>> a.intersection(b)
{4}
Men hur gör man om a och b skulle vara två csv filer?
Filer är ingen datatyp i Python. Har du en lista av listor eller något annat?
Modulen csv
förenklar hantering av CSV-filer. Exempel:
import csv
def read_set_from_csv_file(filename):
result = set([])
with open(filename) as csv_file:
csv_reader = csv.reader(csv_file, delimiter=',')
for row in csv_reader:
for value in list(map(int, row)):
result.add(value)
return result
a = read_set_from_csv_file('read_csv_a.csv')
b = read_set_from_csv_file('read_csv_b.csv')
print(a.intersection(b))