Pandas DataFrame 操作:提取第三列数据
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 的第三列数据。
原文地址: https://www.cveoy.top/t/topic/ocgN 著作权归作者所有。请勿转载和采集!