python生成随机dataframe,第一列为cart,第二列为fav,第三列为pv,第四列如果cart列不为零则为1,否则为0,第五列如果pv列不为零则为1,否则为0
时间: 2023-09-30 16:11:27 浏览: 195
你可以使用Python中的pandas和numpy库来生成随机的dataframe,并使用条件语句来设置第四列和第五列的值。
以下是一个可能的实现:
```python
import pandas as pd
import numpy as np
# 生成随机数据
data = {'cart': np.random.randint(0, 10, size=100),
'fav': np.random.randint(0, 10, size=100),
'pv': np.random.randint(0, 10, size=100)}
# 创建dataframe
df = pd.DataFrame(data)
# 设置第四列和第五列的值
df['cart_flag'] = np.where(df['cart'] > 0, 1, 0)
df['pv_flag'] = np.where(df['pv'] > 0, 1, 0)
# 打印前5行
print(df.head())
```
输出结果如下:
```
cart fav pv cart_flag pv_flag
0 8 1 2 1 1
1 5 4 9 1 1
2 7 8 7 1 1
3 6 9 1 1 1
4 2 2 4 1 1
```
阅读全文