如何提取dataframe中给定几行

在pandas中，可以使用`.loc`或`.iloc`方法来提取DataFrame中的指定行。其中，`.loc`方法使用行标签来进行选择，`.iloc`方法使用行索引来进行选择。

假设我们有以下的DataFrame：

import pandas as pd

df = pd.DataFrame({
    'A': [1, 2, 3, 4, 5],
    'B': ['a', 'b', 'c', 'd', 'e'],
    'C': [0.1, 0.2, 0.3, 0.4, 0.5]
})

print(df)

输出：

   A  B    C
0  1  a  0.1
1  2  b  0.2
2  3  c  0.3
3  4  d  0.4
4  5  e  0.5

如果我们想要提取第2行和第4行，可以使用以下代码：

# 使用.loc方法
rows = [1, 3]
df_selected = df.loc[rows]
print(df_selected)

输出：

   A  B    C
1  2  b  0.2
3  4  d  0.4

如果我们想要提取第2行和第4行的所有列，可以使用以下代码：

# 使用.iloc方法
rows = [1, 3]
df_selected = df.iloc[rows, :]
print(df_selected)

输出：

   A  B    C
1  2  b  0.2
3  4  d  0.4

在上面的代码中，我们分别使用了`.loc`和`.iloc`方法，将需要提取的行的位置存储在一个列表中，然后将其传递给相应的方法。由于需要提取的行是不连续的，所以我们需要传递一个列表。如果需要提取的行是连续的，可以直接传递一个切片对象。