这篇文章持续记录有关dataframe的操作,防止自己忘记
目录
输出dataframe的列名
data.columns.values.tolist()
查找dataframe某一列值得下标
import pandas as pd
data = pd.DataFrame({'a':[1,2,3],'b':[2,3,4]})
data[data.a==1]
输出的结果为:
>>> data[data.a==1]
# 输出所在的行
a b
0 1 2
>>> data[data.a==1].index
# 输出下标
Int64Index([0], dtype='int64')
>>> data[data.a==1].index.tolist()
# 转为list
[0]