python - Pandas 不会创建 .csv 文件
问题描述
我最近开始涉足算法交易并为加密交易构建一个机器人。
为此,我使用 pandas 创建了一个回测器,以使用不同的参数运行不同的策略。我使用的数据集(csv 文件)相当大(每个大约 40mb)。
这些已处理,但只要我想将处理后的数据保存到 csv,什么都不会发生。没有任何输出,甚至没有错误消息。我尝试使用完整路径,尝试仅使用文件名保存它,我什至尝试将其保存为 .txt 文件。似乎没有任何效果。我还尝试了在 stackoverflow 上找到的解决方案。
我正在使用 Anaconda3 以防万一这可能是我的问题的根源。
在这里您可以找到我的代码部分,它试图将数据框保存到文件中。
results_df = pd.DataFrame(results)
results_df.columns = ['strategy', 'number_of_trades', "capital"]
print(results_df)
for i in range(2, len(results_df)):
if results_df.capital.iloc[i] < results_df.capital.iloc[0]:
results_df.drop([i],axis="index")
#results to csv
current_dir = os.getcwd()
results_df.to_csv(os.getcwd()+'\\file.csv')
print(results_df)
谢谢您的帮助!
解决方案
您可以大大简化您的代码并将其编写为(也应该运行得更快):
results_df = pd.DataFrame(results)
results_df.columns = ['strategy', 'number_of_trades', "capital"]
print(results_df)
first_row_capital= results_df.capital.iloc[0]
indexer_capital_smaller= results_df.capital < first_row_capital
values_to_delete= indexer_capital_smaller[indexer_capital_smaller].index
results_df.drop(index=values_to_delete, inplace=True)
#results to csv
current_dir = os.getcwd()
results_df.to_csv(os.getcwd()+'\\file.csv')
print(results_df)
我认为,您的代码中的主要问题可能是,每次您在数据框中找到满足条件的条目时都编写 csv,并且只有在找到这种情况时才编写它。
如果您只是删除 csv 输出但不再需要内存中的数据帧,则可以使其更简单:
results_df = pd.DataFrame(results)
results_df.columns = ['strategy', 'number_of_trades', "capital"]
print(results_df)
first_row_capital= results_df.capital.iloc[0]
indexer_capital_smaller= results_df.capital < first_row_capital
#results to csv
current_dir = os.getcwd()
results_df[indexer_capital_smaller].to_csv(os.getcwd()+'\\file.csv')
print(results_df[indexer_capital_smaller])
第二个变体仅在写入过滤后的行和打印内容之前应用过滤器。
推荐阅读
- variables - Unified name for the national identification number
- python - Python:对象列表与整数列表行为
- ruby-on-rails - 使用 devise 作为身份验证方法在 Rails 上隐藏某些用户信息的最佳方法?
- r - 为什么有人应该使用 {} 在 R 中初始化一个空对象?
- node.js - Node.JS HTTP/HTTPS/etc 通过代理隧道请求
- reactjs - Material Ui - 在“选择”模式下将所需属性添加到 TextField
- javascript - Json 项目的计数器
- html - 旋转社交媒体图标-引导程序 4
- javascript - JS 动态隐藏/删除具有相同类名的 div
- javascript - 信息框上的 CSS 悬停效果但颜色不同