Я пытаюсь выбрать столбцы данных из кадра данных csv, используя список столбцов, а затем сохранить извлеченные данные в другой файл csv_file.
columns_to_extract = model.columns # 'Name', 'Age', 'Address', 'Sex' FORM the model database
df = pd.read_csv('some-data.csv', encoding='ISO-8859-1')
df = df[[columns_to_extract]] # this doesn't work
# also tried df = df[[columns_to_extract.split(',)]] and didn't work too
import StringIO
s_df = StringIO.StringIO()
df.to_csv(s_df)