pandas get names of columns with only certain code example
Example: select specific column names from dataframe
import pandas
data = pandas.DataFrame({'A' : ['X', 'Y'],
'B' : 1,
'C' : [2, 3]})
print data[['A', 'B']]
# Output A B
# 0 X 1
# 1 Y 1