Θέλω να λάβω μια λίστα με τις επικεφαλίδες των στηλών από ένα πλαίσιο δεδομένων pandas DataFrame. Το DataFrame θα προέλθει από την είσοδο του χρήστη, οπότε δεν ξέρω πόσες στήλες θα υπάρχουν ή πώς θα ονομάζονται.
Για παράδειγμα, αν μου δοθεί ένα DataFrame όπως αυτό:
>>> my_dataframe
y gdp cap
0 1 2 5
1 2 3 9
2 8 7 2
3 3 4 7
4 6 7 7
5 4 8 3
6 8 2 8
7 9 9 10
8 6 6 4
9 10 10 7
Θα ήθελα να λάβω μια λίστα όπως αυτή:
>>> header_list
['y', 'gdp', 'cap']
Μπορείτε να λάβετε τις τιμές ως λίστα κάνοντας:
list(my_dataframe.columns.values)
Επίσης, μπορείτε απλά να χρησιμοποιήσετε: (όπως φαίνεται στην απάντηση Ed Chum's answer):
list(my_dataframe)
Αυτό είναι διαθέσιμο ως my_dataframe.columns
.