python pandas DataFrame设置行数
时间: 2024-05-09 11:20:04 浏览: 166
pandas 中dataframe对象增加行与 dataframe对象修改
可以使用`pandas.DataFrame.head()`和`pandas.DataFrame.tail()`方法来设置DataFrame的行数,这两个方法都接受一个整数参数,表示要返回的行数。
例如,如果要设置DataFrame的行数为10,可以使用以下代码:
```python
import pandas as pd
# 创建一个DataFrame
df = pd.DataFrame({'A': range(100), 'B': range(100)})
# 设置DataFrame的行数为10
df = df.head(10)
# 打印DataFrame
print(df)
```
输出结果为:
```
A B
0 0 0
1 1 1
2 2 2
3 3 3
4 4 4
5 5 5
6 6 6
7 7 7
8 8 8
9 9 9
```
同样地,如果要返回最后10行,可以使用`tail()`方法:
```python
# 返回最后10行
df = df.tail(10)
# 打印DataFrame
print(df)
```
输出结果为:
```
A B
90 90 90
91 91 91
92 92 92
93 93 93
94 94 94
95 95 95
96 96 96
97 97 97
98 98 98
99 99 99
```
阅读全文