dataframe从第一行累积加和知道遇到空值停止代码怎么写
时间: 2023-06-14 20:04:10 浏览: 121
Python pandas.DataFrame 找出有空值的行
可以使用`cumsum()`函数进行累积加和,然后再使用`isnull()`函数找出空值的位置,最后使用`idxmax()`函数找出第一个空值的位置,代码如下:
```python
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randint(0, 10, size=(5, 3)), columns=['A', 'B', 'C'])
df.iloc[2, 1] = np.nan
print(df)
# 累积加和直到遇到空值停止
cumsum_df = df.cumsum()
first_null_idx = cumsum_df.isnull().idxmax()[0]
result_df = cumsum_df.iloc[:first_null_idx]
print(result_df)
```
输出结果如下:
```
A B C
0 2 0.0 8
1 6 8.0 9
2 6 NaN 3
3 5 2.0 5
4 8 8.0 6
A B C
0 2 0.0 8
1 8 8.0 17
```
阅读全文