Python 将csv文件的index处理后得到了几个列表，然后想要按列表中的元素为index从原csv文件中查找其在某一列的值

Question

tmp保存csv的信息，如下图所示

然后按照f1的大小将其排序：

score = tmp['f1'].order()
score.head()
len(score)

然后将整个数据5等分：

df = {}
close = tmp['close']
port1 = list(score.index)[: len(score)//5]
port2 = list(score.index)[ len(score)//5: 2*len(score)//5]
port3 = list(score.index)[ 2len(score)//5: -2len(score)//5]
port4 = list(score.index)[ -2*len(score)//5: -len(score)//5]
port5 = list(score.index)[ -len(score)//5: ]

想要按port的不同，分别获取其中stock的close的值进行相应的计算

def caculate_port_daily_return(port,start,end,close):

for stock in port

如何获取？请指教~谢谢~~~

Answer 1

-------------------------------
port1 = list(score.index)[: len(score)//5]
port2 = list(score.index)[ len(score)//5: 2*len(score)//5]
port3 = list(score.index)[ 2len(score)//5: -2len(score)//5]
port4 = list(score.index)[ -2*len(score)//5: -len(score)//5]
port5 = list(score.index)[ -len(score)//5: ]
--------------------------------

from collections import deque

# deque([f1, f2, f3, f4, f5, f6, f7, f8, f9, f10])
list_ = deque(score.index)
eq = 5
el = len(score)//eq  # = 2

#     1         2         3         4         5    -> eq (然后将整个数据5等分)
#[[f1, f2], [f3, f4], [f5, f6], [f7, f8], [f9, f10]]
port = [[list_.popleft() for _ in range(el)] for _ in range(eq)]

i don't have any idea about "def calculate_port_daily_return"
check it plz: https://www.quantopian.com/posts/how-to-calculate-daily-return