Как я могу разделить файл csv на два файла на основе значений в столбце в Python?
У меня есть файл csv с 22 столбцами и X строк. Как я могу разделить файл csv на 2 отдельных файла в зависимости от того, какое значение находится, например. Столбец "item_number"?
Так, например, если строка имеет «1234» или «2345» в столбце «item_number», я хочу, чтобы она (вся строка со всеми соответствующими столбцами) попала в csv_file_1. Если строка имеет «5678» или «6789» в столбце «item_number», я хочу, чтобы она попала в csv_file_2.
Я хочу, чтобы код перебирал все X строк в исходном входном CSV-файле.
Таким образом, на выходе должны быть два отдельных файла csv, отсортированных в зависимости от конкретных значений «item_number».
Заранее спасибо за вашу помощь, и, пожалуйста, дайте мне знать, если потребуется дополнительная информация!
Ответы 1
Это то, что вы ищите ?
import pandas
data = pandas.read_csv('mycsv.csv')
# just make sure to remove the quotation if the numbers are not string
csv2 = data[data['item_number'].isin(['5678','6789'])]
csv1 = data[data['item_number'].isin(['1234','2345']]
csv1.to_csv('csv1.csv', index=False)
csv2.to_csv('csv2.csv', index=False)
Приведенный выше код работает. Сначала вы загружаете CSV-файл в фрейм данных, а затем применяете фильтр элементов и сохраняете его в соответствующих CSV-файлах.
Это идеально! Большое спасибо всем!!
@user8192647 user8192647, если это отвечает на ваш вопрос, не могли бы вы принять ответ. Спасибо.
О, извините, я никогда раньше не использовал Stack Overflow — ответ принят! Еще раз спасибо!
о, не беспокойтесь, рад, что помог.
Итак, что вы пробовали до сих пор?