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

Tags:

Misc Example