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

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)