pandas提供了query语法,能用于复杂的操作,如过滤等
import pandas as pd
data = pd.read_csv("C:\\Users\\Administrator\\Desktop\\文档\\作业\\云计算与大数据实验一\\云计算与大数据实验一\\600519.csv", encoding='gb2312')
def find1():
print("*************显示某个输入的日期的当日数据****************")
a = input("输入日期按年-月-日的格式 ")
info = data[data['日期'] == str(a)]
print(info)
def find2():
print("**************显示收盘价高于某个输入的价格的所有日期********")
b = input("输入 ")
b = int(b)
info = data.query(' 收盘价 > @b')
print(info)
def find3():
print("**************显示股价波动高于某个输入的值的所有日期********")
c = input("输入 ")
c = float(c)
info2 = data.query('(最高价-最低价)> @c')
print(info2['日期'])
def find4():
print("**************显示成交量高于某个输入的值的所有日期*********")
d = input("输入 ")
d = int(d)
info3 = data.query('成交量> @d')
print(info3['日期'])
def find5():
print("***************显示成交额高于某个输入的值的所有日期********")
e = input("输入 ")
e = int(e)
info4 = data.query('成交金额> @e')
print(info4['日期'])
while True:
print(" 显示某个输入的日期的当日数据 输入1")
print(" 显示收盘价高于某个输入的价格的所有日期 输入2")
print(" 显示股价波动高于某个输入的值的所有日期 输入3")
print(" 显示成交量高于某个输入的值的所有日期 输入4")
print(" 显示成交额高于某个输入的值的所有日期 输入5")
a =int(input("请输入选择 "))
if a==1:
find1()
if a ==2:
find2()
if a ==3:
find3()
if a ==4:
find4()
if a ==5:
find5()