如何将主题列表(从gensim lda get_document_topics())更改为DataFrame格式

pip install pandas

import pandas as pd

# 假设你已经使用gensim的LDA模型获取了文档的主题列表
# doc_topics_list 是一个包含多个文档主题列表的列表
# 每个文档的主题列表是一个包含 (topic_id, topic_probability) 元组的列表

# 示例数据
doc_topics_list = [
    [(0, 0.1), (1, 0.4), (2, 0.5)],
    [(0, 0.3), (1, 0.2), (2, 0.5)],
    [(0, 0.2), (1, 0.6), (2, 0.2)]
]

# 将主题列表转换为DataFrame格式
data = {
    'doc_id': [],  # 文档ID，可根据实际情况添加
    'topic_id': [],
    'probability': []
}

for idx, topics in enumerate(doc_topics_list):
    for topic_id, probability in topics:
        data['doc_id'].append(idx)
        data['topic_id'].append(topic_id)
        data['probability'].append(probability)

df = pd.DataFrame(data)

print(df)