python - 在数据框架中寻找所选数据类型列的位置。
这是我的数据框架。
c_id fname age salary lname
1 abc 21 21.22 yyy
2 def 41 23.4 zzz
我需要显示列名相对于数据类型的位置,所以我的输出应该是:
**FOR INT:**
col_name position
c_id 0
age 2
**for str:**
col_name position
fname 1
lname 4
**for float:**
col_name position
salary 3
【回答】:
IIUC,你可以直接从dtypes中创建一个数据框架,然后重置索引来获得位置索引号。
col_df = (
pd.DataFrame(df.dtypes, columns=["DataType"])
.rename_axis("Column")
.reset_index()
.rename_axis("Position")
)
Column DataType
Position
0 c_id int64
1 fname object
2 age int64
3 salary float64
4 lname object
print(col_df[col_df['DataType'] == 'object'])
Column DataType
Position
1 fname object
4 lname object
【回答】:你可以在DataFrame上使用.dtypes方法。
df.dtypes
>>> Name object
Age int64
City object
Marks float64
dtype: object
df.dtypes['Name']
>>> object