Удаление словаря из списка

Question

У меня есть файл CSV, который я читал как список словарей для каждой строки. Я хочу удалить все записи в списке с адресом электронной почты ''. Я пробовал:

#!/usr/bin/python 
import csv 


def import_users(location_of_file): 
    with open(location_of_file, 'r', newline='', encoding='utf-8-sig') as openfile: 
     reader = csv.DictReader(openfile) 
     for row in reader: 
      yield row 

def save_csv(data, location): 
    with open(location, 'w', newline='', encoding='utf-8-sig') as file: 
     fieldnames = ['EmailAddress', 'GivenName', 'Surname', 'Company', 'Department'] 
     writer = csv.DictWriter(file, fieldnames=fieldnames) 
     writer.writeheader() 
     for item in data: 
      writer.writerow(item) 



if __name__ == '__main__': 
    users = list(import_users('C:\Temp\Example.csv')) 
    for user in users: 
     if user['EmailAddress'] == '': 
      del user 
     else: 
      pass 
    save_csv(users, 'C:\Temp\Output.csv') 

Но мои результаты по-прежнему имеют записи без адреса электронной почты. Что я делаю не так?

Answer 1

Не изменяйте элементы списка, пока они раздражают их.

Вместо

for user in users: 
    if user['EmailAddress'] == '': 
     del user 
    else: 
     pass 

ли

users = filter(lambda user: user['EmailAddress'] != '', users) 

Answer 2

Вы, вероятно, будет лучше, если новый список, а не удаление элементов:

users = [user for user in users if user['EmailAddress'] != ''] 

Answer 3

Итерация по структуре данных, изменяемого плохая практика (и приведет к супер раздражающих ошибок). Поэтому вы должны составить другой список, содержащий только те элементы, которые вы хотите. Вы можете сделать это с помощью цикла:

users = list(import_users('C:\Temp\Example.csv')) 
filtered_users = [] 
for user in users: 
    if user['EmailAddress'] == '': 
     filtered_users.append(user) 
save_csv(filtered_users, 'C:\Temp\Output.csv') 

Или используя питон функцию фильтра:

users = list(import_users('C:\Temp\Example.csv')) 
filtered_users = filter(lambda user: user.get('EmailAddress') != '', users) 
save_csv(filtered_users, 'C:\Temp\Output.csv') 

Answer 4

Вам не нужно когда-либо создать список в памяти. Вы можете обойти генераторы и итераторы вместо:

if __name__ == '__main__': 
    users = import_users('C:/Temp/Example.csv') 
    save_csv((user for user in users if user['EmailAddress'] != ''), 
      'C:/Temp/Output.csv') 

Answer 5

Вот решение, использующее pandas:

Пример данных:

import pandas as pd 
#Read csv data 
df = pd.read_csv('data.csv') 
#Get only the rows having email address 
dfo = df[pd.notnull(df['EmailAddress'])] 
#Save to a file 
dfo.to_csv('output.csv', index=False) 

Пример вывода: