Как мне написать несколько списков в свой собственный столбец в текстовом файле? - PullRequest
1 голос
/ 30 января 2020

У меня есть пять списков, которые мне нужно составить в виде текстового файла, каждый в отдельном столбце. Пока у меня есть

with open("PR.txt","w") as f:
    PR = [[Velocity], [Angle], [Impact], [y], [Distance]]
    for (x) in zip(PR):
        f.write("{0}\t{1}\t{2}\t{3}\t{4}\n".format(*x))

Я хочу, чтобы он написал текстовый файл, который идет

Velocity Angle Impact y Distance
Velocity Angle Impact y Distance
Velocity Angle Impact y Distance

и так далее

Я не могу понять, как это сделать .

Ответы [ 2 ]

1 голос
/ 30 января 2020

Если у вас есть все пять списков одинаковой длины,

with open("PR.txt","w") as f:
 f.write("Velocity\tAngle\tImpact\ty\tDistance") 
 for i in range(0, len(Velocity)):
    # Velocity here is the list
    f.write("{0}\t{1}\t{2}\t{3}\t{4}\n".format(Velocity[i],Angle[i], Impact[i], y[i], Distance[i]))
0 голосов
/ 30 января 2020

Вы можете использовать * args для распаковки столбцов в zip:

# test input
pr = [['v 0', 'v 1', 'v 2', 'v 3', 'v 4'], ['10', '11', '12', '13', '14'], ['0', '1', '2', '3', '4'], ['y0', 'y1', 'y2', 'y3', 'y4'], ['dist 0', 'dist 1', 'dist 2', 'dist 3', 'dist 4']]

with open('file.txt', 'w') as fh:
    cols = ['Velocity', 'Angle', 'Impact', 'y', 'Distance']
    fh.write('\t'.join(cols) + '\n')

    # here is where you unpack everything
    for row in zip(*pr):
        fh.write('\t'.join(row) + '\n')

Какие выходные данные

Velocity    Angle   Impact  y   Distance
v 0         10      0       y0  dist 0
v 1         11      1       y1  dist 1
v 2         12      2       y2  dist 2
v 3         13      3       y3  dist 3
v 4         14      4       y4  dist 4
...