csv抽取第一行和第二行形成一个新的数组python语言

原创于 2019-12-20 16:34:34 发布 · 623 阅读

CC 4.0 BY-SA版权

文章标签：

本文介绍如何使用Python的Pandas库从CSV文件中读取数据，并演示了如何抽取特定行形成新的数组。通过具体代码示例，展示了数据处理的基本流程。

P.csv 数据如下：

year,staock,age
1,2,3
4,5,6
3,5,6
55,67,88

目的让Pcsv抽取第一行和第二行形成一个新的数组。

结果：[ [1 2 3]

[4,5,6]

代码如下：

import pandas as pd

df = pd.read_csv(r'P.csv')

data1 = df.values[0, :]
data12 = df.values[1, :]

data=np.vstack((data1,data12))
print(data)