如何用Pandas从数据集中快速创建单行DataFrame
快速提取单行生成独立DataFrame的方法
不用手动构造字典,直接基于原DataFrame就能实现,分两种场景处理:
指定行提取
按行位置提取(索引从0开始)
用iloc配合列表索引,直接返回DataFrame格式:
import pandas as pd df = pd.read_csv('filename.csv') # 提取第3行(对应索引2) df2 = df.iloc[[2]]
按行标签提取
如果你的DataFrame有自定义索引(比如某列是行标签),用loc:
# 提取标签为"target_row"的行 df2 = df.loc[["target_row"]]
随机行提取
用sample()方法直接随机选一行,返回结果就是DataFrame:
# 随机抽取1行 df2 = df.sample(n=1) # 如果需要固定随机结果(可复现),加random_state参数 df2 = df.sample(n=1, random_state=42)
注意:别直接写df.iloc[2]或者df.loc["target_row"],这样返回的是Series对象,不是DataFrame。用列表包裹索引/标签,就能直接得到带原列名的单行DataFrame。
内容的提问来源于stack exchange,提问作者prof31
相关产品推荐
相关产品推荐

