Чувствительный к регистру CSV-запрос в python - PullRequest
0 голосов
/ 22 сентября 2018

У меня есть набор данных для столкновений животных на Вирджинских дорогах.Я пытаюсь извлечь все столкновения с участием медведей.В моем наборе данных медведь набирается, по крайней мере, с тремя различными комбинациями падежей (т. Е. Bear, bear, BEAR).Ниже приведен код, который я использую для извлечения столкновений медведей:

import csv

with open('google.csv', 'r') as f:
    reader = csv.reader(f)
    lines = [ row for row in reader if "bear" in row[4] ]

with open('outfile.csv', "w") as outfile:
    writer = csv.writer(outfile)
    writer.writerows(lines)

Как я могу сделать это без учета регистра?

1 Ответ

0 голосов
/ 22 сентября 2018

Попробуйте это:

import csv

with open('google.csv', 'r') as f:
    reader = csv.reader(f)
    lines = [ row for row in reader if "bear" in row[4].strip().lower() ]

with open('outfile.csv', "w") as outfile:
    writer = csv.writer(outfile)
    writer.writerows(lines)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...