pandas习题 108：select_dtypes() 按类型筛选数据

看过来

《pandas 教程》持续更新中，提供建议、纠错、催更等加作者微信: gr99123（备注：pandas教程）和关注公众号「盖若」ID: gairuo。跟作者学习，请进入 Python学习课程。欢迎关注作者出版的书籍：《深入浅出Pandas》和《Python之光》。

（编码）行政部把员工信息汇总成一张小表：

import pandas as pd

df = pd.DataFrame({
    '工号': [1001, 1002, 1003],                       # int64
    '姓名': ['张三', '李四', '王五'],                    # object (str)
    '年龄': [28, 35, 29],                             # int64
    '入职日期': pd.to_datetime(['2022-03-15', '2021-07-01', '2023-01-10']),  # datetime64[ns]
    '交通补贴': [200, 150.5, 180],                     # float64
    '备注': ['地铁', '自驾', '公交']                      # object (str)
})

任务：

请用一行 select_dtypes() 完成以下 4 个独立小题：

选出 所有整数列（int64）。
选出 所有字符串列（object）。
选出 所有数值型列（int64 + float64，排除日期和字符串）。
选出 所有时间类型列（datetime64[ns]）。

参考答案

代码如下：

# 1. 整数列
df.select_dtypes('int64')

# 2. 字符串列
df.select_dtypes('object')

# # 3. 数值列（int64 + float64）
df.select_dtypes(include=['int64', 'float64'])

# # 4. 时间列
df.select_dtypes('datetime64[ns]')

（完）

pandas习题 108：select_dtypes() 按类型筛选数据

参考答案

相关内容