Как записать фрейм данных / список в csv-файл? - PullRequest
Новая
       12

Как записать фрейм данных / список в csv-файл?

2 голосов
/ 18 февраля 2020

Я написал следующий код и хочу записать вывод в файл csv. Как я могу это сделать? Спасибо за помощь. Вот выдержка из моего кода: 

columnNames = ['Stride Length', 'Stand Duration', 'Swing Duration', 'Douple Support Time', 'Relation Swing Stand']

resultsPerRowRatio = list()

for i in range(len(stepData)):
     stepDataLeft = stepData.to_numpy()[i, 0:5]
     stepDataRight = stepData.to_numpy()[i, 5:10]
     dataF = pd.DataFrame({'stepDataLeft': stepDataLeft, 'stepDataRight': stepDataRight}, index=columnNames)

     #Symmetry Ratio:
     dataF['Symmetry Ratio Row']=np.where(dataF['stepDataLeft'] < dataF['stepDataRight'],sv.symmetryRatio(dataF['stepDataLeft'],dataF['stepDataRight']), sv.symmetryRatio(dataF['stepDataRight'],dataF['stepDataLeft']))
     resultsPerRowRatio.append(dataF)

print("Symmetry Ratio: "+str(resultsPerRowRatio))

Это мой вывод:

enter image description here

Я хочу написать это в CSV-файл. Но я не знаю как. Я уже пробовал разные варианты.

1 Ответ

2 голосов
/ 18 февраля 2020

Это подход к рассмотрению использования to_csv():

to_csv() Параметры (доступно больше, но это простой случай):

  • header: заголовок столбца, True для включения или False для исключения режима
  • : «a» для добавления в существующий файл, «w» для создания и записи нового файла
  • index: True для включения индексации, False для exclude

Пример: 

PATH = your_folder_path
FILENAME = output_file_name

if os.path.isfile(PATH+FILENAME):
    dataF.to_csv(PATH+FILENAME, header=False, mode='a', index=True)
else:
    dataF.to_csv(PATH+FILENAME, header=True, mode='w', index=True)

Эта структура выполнит проверку logi c, чтобы увидеть, существует ли ваш выходной файл в пути к папке, которую вы Определив, если он существует, он добавит данные в файл csv без соответствующих имен столбцов заголовка с помощью вызова to_csv (). Если он не существует, он сгенерирует новый CSV-файл и включит имена столбцов в качестве заголовка.

...