df_stock为dataframe 当high_winner大于95%时,且buy_signal的值为0时,则sell_signal的值为-1,否则为0
时间: 2024-04-09 12:32:36 浏览: 22
感谢您的补充说明。根据您的描述,我已经更新了代码以符合您的要求。以下是更新后的代码示例:
```python
import pandas as pd
# 读取数据集
df_stock = pd.read_csv("your_data.csv", parse_dates=["date"])
df_stock.set_index("date", inplace=True)
# 新增字段 bought 和 sold,初始值为 0
df_stock["bought"] = 0
df_stock["sold"] = 0
# 遍历数据集
for index, row in df_stock.iterrows():
low_winner = row["low_winner"]
high_winner = row["high_winner"]
buy_signal = 0
sell_signal = 0
# 判断买入信号
if low_winner < 0.005:
buy_signal = 1
# 判断卖出信号
if high_winner > 0.95:
if buy_signal == 0:
sell_signal = -1
# 更新买入和卖出字段
if buy_signal == 1 and row["bought"] == 0:
df_stock.at[index, "bought"] = 1
if sell_signal == -1 and row["bought"] == 1 and row["sold"] == 0:
df_stock.at[index, "sold"] = -1
# 忽略买入信号
if row["sold"] == -1:
df_stock.at[index, "buy_signal"] = 0
# 打印更新后的数据集
print(df_stock)
```
运行此代码将更新数据集中的 "bought" 和 "sold" 字段,并在最后打印更新后的数据集。根据您的要求,当 "high_winner" 大于95% 且 "buy_signal" 为0 时,"sell_signal" 将设置为-1。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)