¿Cómo transponer un conjunto de datos en un archivo CSV?
Pregunta
Por ejemplo, me gustaría transformar:
Name,Time,Score
Dan,68,20
Suse,42,40
Tracy,50,38
Dentro:
Name,Dan,Suse,Tracy
Time,68,42,50
Score,20,40,38
EDITAR: La pregunta original utilizó el término "transposición" incorrectamente.
Solución
Si todo el contenido del archivo encaja en la memoria, puede usar
import csv
from itertools import izip
a = izip(*csv.reader(open("input.csv", "rb")))
csv.writer(open("output.csv", "wb")).writerows(a)
Básicamente puedes pensar en zip()
y izip()
Como operaciones de transposición:
a = [(1, 2, 3),
(4, 5, 6),
(7, 8, 9)]
zip(*a)
# [(1, 4, 7),
# (2, 5, 8),
# (3, 6, 9)]
izip()
Evita la copia inmediata de los datos, pero básicamente hará lo mismo.
Otros consejos
Transferido de input.csv
a output.csv
. Los pandas también pueden ayudar.
import pandas as pd
pd.read_csv('input.csv', header=None).T.to_csv('output.csv', header=False, index=False)
from itertools import izip
from csv import reader, writer
with open('source.csv') as f, open('destination.csv', 'w') as fw:
writer(fw, delimiter=',').writerows(izip(*reader(f, delimiter=',')))
Si lines
es la lista de su texto original de lo que debería ser
for i in range(1,len(lines)):
lines[i] = lines[i].split(',')
new_lines = []
for i in range(len(lines[0])):
new_lines.append("%s,%s,%s" % (lines[0][i], lines[1][i], lines[2][i]))
o usar csv
Módulo de Python - http://docs.python.org/library/csv.html
La misma respuesta de Nosklo (todos los créditos para él), pero para Python3:
from csv import reader, writer
with open('source.csv') as f, open('destination.csv', 'w') as fw:
writer(fw, delimiter=',').writerows(zip(*reader(f, delimiter=',')))
La forma más simple es:
import numpy as np
import pandas as pd
_mat = pd.read_csv("test.csv")
_mat = _mat[_mat.columns[0:3]].values
_t_mat = np.transpose(_mat)
Resultado:
- La matriz de entrada es: [[1 2 3] [4 5 6]
- La salida es: [[1 4] [2 5] [3 6]
Lea el CSV en pandas
Marco de datos, Pandas tiene una función de construcción para la transposición que se puede invocar a continuación.
import pandas as pd
csv = pd.read_csv("test.csv", skiprows=1)
# use skiprows if you want to skip headers
df_csv = pd.DataFrame(data=csv)
transposed_csv = df_csv.T
print(transposed_csv)