Как включить переменные цикла for в файл output.csv - PullRequest
0 голосов
/ 17 октября 2018

Я новичок в программировании на Python и пытаюсь выяснить, как я могу получить желаемый результат из кода.У меня есть цикл for внутри другого цикла for.Я стремлюсь экспортировать файл .csv, который включает в себя результаты моего кода помимо сведений о тех переменных цикла for, которые соответствуют моим результатам.Мне интересно, как я могу манипулировать to.csv, чтобы получить то, что мне нужно (пожалуйста, найдите ниже мой желаемый файл output.csv).

примерный скрипт

df1 = [1, 2, 3, 4, 5]
df2 = [I, II, III]
for i in df1.index:
    for j in df2.index:
        ### bunch of calculations happening here resulted in output###
output.to_csv(r'd:\project\output.csv')

мой желаемый output.csvфайл

output     df1      df2
A          1        I
B          1        II
C          1        III
D          2        I
E          2        II
F          2        III
G          3        I
H          3        II
I          3        III
G          4        I
K          4        II
L          4        III
N          5        I
M          5        II
O          5        III

1 Ответ

0 голосов
/ 17 октября 2018

Попробуйте это (панды):

import pandas as pd,string
l=[]
l2=[]
df1 = [1, 2, 3, 4, 5]
df2 = ['I', 'II', 'III']
for i in df1:
    for j in df2:
        l.append(i)
        l2.append(j)
output=pd.DataFrame({'output':string.ascii_uppercase[:len(l)],'df1':l,'df2':l2})
output.to_csv(r'd:\project\output.csv')
...