Pandas DataFrame 操作:提取第三列数据

以下代码展示了如何使用 Pandas 库读取 CSV 文件并提取 DataFrame 第三列数据:

stock_sales = pd.read_csv(
    comp_dir / 'train.csv',
    usecols=['Data_ID', 'Time', 'Close_Price'],
    dtype={
        'Data_ID': 'int',
        #'Time': 'datetime64',
        'Close_Price': 'float32',
    },
    parse_dates=['Time'],
    infer_datetime_format=True,
)
x = stock_sales.copy()

# 提取第三列数据
third_column = x.iloc[:, 2]  # 第三列的索引是2,使用iloc方法取出

stock_sales 的格式:

stock_sales 是一个 Pandas DataFrame 格式的数据,包含三列:

  • 'Data_ID'(整型)
  • 'Time'(日期时间型)
  • 'Close_Price'(浮点型)

说明:

  • iloc 方法用于通过索引来访问 DataFrame 中的特定行或列。
  • [:, 2] 表示选择所有行,以及索引为 2 的列(即第三列)。
  • x.iloc[:, 2] 赋值给 third_column 变量,就可以访问 DataFrame 的第三列数据。
Pandas DataFrame 操作:提取第三列数据

原文地址: https://www.cveoy.top/t/topic/ocgN 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录