how to get a list of column names in pandas code example
Example 1: pd dataframe get column names
lst = data.columns.values # data is dataframe
Example 2: how to find columns of a dataframe
list(my_dataframe.columns.values)
Example 3: get column names as list
public static List getColumnNames(){
List columnList = new ArrayList<>();
try {
ResultSetMetaData rsMetaData = rs.getMetaData();
for (int colNum = 1; colNum < getColumnCount() ; colNum++) {
columnList.add(rsMetaData.getColumnLabel(colNum));
}
} catch (SQLException e) {
System.out.println("ERROR WHILE GETTING COLUMN NAMES "+ e.getMessage());
}
return columnList;
}
Example 4: pass list of columns names in pandas
>>> import pandas
>>> # create three rows of [0, 1, 2]
>>> df = pandas.DataFrame([range(3), range(3), range(3)])
>>> print df
0 1 2
0 0 1 2
1 0 1 2
2 0 1 2
>>> my_columns = ["a", "b", "c"]
>>> df.columns = my_columns
>>> print df
a b c
0 0 1 2
1 0 1 2
2 0 1 2