9阅网

您现在的位置是:首页 > 知识 > 正文

知识

python - 在数据框架中寻找所选数据类型列的位置。

admin2022-11-06知识17

这是我的数据框架。

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