使用带有 Excel 工作表数据的 Pandas 获取基于用户 ID 的数据？

Question

我想根据 u_id 获取特定数据以绘制图表和预测市场汇率。例如：我的 excel 表中有 1000 个数据，U_ID 为 5，我想获取 U_ID 的第 5 个完整数据。我的数据格式是..

上面的屏幕截图是我的 excel 格式输入数据，我想根据 U_ID 获取数据。如果 U_ID 为 2 表示我想获取所有数据，包括 P_ID、产品（勺子、杯子）、产品数量、总量，地点，日期。

我试过了，但还是没有得到预期的输出。

import pandas as pd
from pandas import DataFrame as df
from pandas import *
import matplotlib.pyplot as plt
from selenium.webdriver.common import by
excel_file = 'FIRST1.xls'
mov = pd.read_excel(excel_file)
# print(mov)
DataFrame(mov, columns=["SNO",  "U_ID", "P_ID", "SUBPRODUCTS",  "PRODUCT QUANTITY", "TOTAL AMOUNT", "LOCATION", "DATE"])
mov.sort_values(by="U_ID", inplace=True)
# df.sort_values(by='PURCHASE AMOUNT')
print (mov.iloc[1])
getda = mov.iloc[1]
gd = getda[1]
od = getda[4]

它以排序形式提供数据..任何人帮助我.. 提前致谢

Answer 1

试试这个：

import pandas as pd
import matplotlib.pyplot as plt

excel_file = 'FIRST1.xls'
df = pd.read_excel(excel_file)
df.columns = ["SNO",  "U_ID", "P_ID", "SUBPRODUCTS",  "PRODUCT QUANTITY", "TOTAL AMOUNT", "LOCATION", "DATE"]

getda = df[df.U_ID == 2].values # <- .values gives you the values of your data row as a list!

gd = getda[1]
od = getda[4]

如果您有多个满足条件的行，则对结果（即列表列表）进行交互：

for row in getda:
    gd = row[1]
    od = row[4]