Tengo un gl grande del que me gustaría hacer cálculos y predicciones, el problema es que no puedo encontrar una manera de obtener una lista de todos los nombres de columna JUNTO con la columna "número". Es imposible contar desde arriba para ver qué número es, y prefiero no escribir todos los nombres de las columnas.
Sería bueno poder usar algo como esto:
df.iloc[:, np.r_[2, 5:10, 22:102, 109:129]]
pero para que eso funcione, necesito saber qué columna tiene qué número.
list(df) me da una buena lista, pero sin los números, lo que lo hace inútil en esta búsqueda.

1
Mactilda 23 oct. 2019 a las 15:42

1 respuesta

La mejor respuesta

Sugiero crear un diccionario con enumerate:

df = pd.DataFrame({
        'A':list('abcdef'),
         'B':[4,5,4,5,5,4],
         'C':[7,8.0,9,4.0,2,3],
         'D':[1,3,5,7,1,0],
         'E':[5,3,6,9,2,4],
         'F':list('aaabbb')
})

d = dict(enumerate(df))
print (d)
{0: 'A', 1: 'B', 2: 'C', 3: 'D', 4: 'E', 5: 'F'}

O lista de tuplas como @Chris sugerido en los comentarios:

L = list(enumerate(df))
print (L)
[(0, 'A'), (1, 'B'), (2, 'C'), (3, 'D'), (4, 'E'), (5, 'F')]
2
jezrael 23 oct. 2019 a las 13:02