使用 django 從 MySQL 表中獲取超過 25 列的數據 model

Question

我有一個帶有 4 個表（all_users、freq_user、special_users、產品）的 MySQL db (companies_data)。 all_users表有 27 列和數千行，我使用該數據來提供我的機器學習算法。

我在后端使用 python、django rest_framework。

現在我想使用 django model 獲取這些數據並將其轉換為 pandas dataframe 以便我可以預測並將預測添加到其他表中。 有沒有一種簡單的方法可以使用 django model 獲取所有數據？

現在我使用帶有所有列名的原始 SQL 查詢來獲取數據。

Answer 1

你有這些表的 django 模型嗎？ 一般來說，您可以為每個表創建一個 django model，然后為您想要的字段構建查詢集（您也可以過濾），僅此而已。

import pandas as pd
import datetime
from myapp.models import BlogPost


# limit which fields
queryset = User.objects.filter(...).annotate(...).values_list('f1','f2',...)
df = pd.DataFrame(list(queryset), columns=['f1','f2',...])