在Python中将函数导出到.csv或.xlsx文件

Question

我在Python中使用apripori算法，想知道如何将结果导出到excel或csv？

from mlxtend.preprocessing import TransactionEncoder
from mlxtend.frequent_patterns import apriori
apriori(df_one_hot,min_support=0.05,use_colnames=True).sort_values(by='support',ascending=False)

Answer 1

将结果存储在pandas数据框中，并使用to_csv写入CSV文件。

import pandas as pd
results = apriori(df_one_hot,min_support=0.05,use_colnames=True).sort_values(by='support',ascending=False)
results_df = pd.DataFrame(results)
results_df.to_csv('results.csv', index=False)