python 分类前百分之十_Python Pandas计算每个类别的回报百分比

我使用了每个访问者的索引值，并检查了下一年相同的索引值(也就是相同的vistor_ID)是否大于0。然后以True或False的形式将其添加到字典中，您可以将其用于条形图。我还列出了两个列表(times\u return和return-at-u-all)，用于额外的数据操作。在import pandas as pd

# Part 1, Building the dataframe.

df = pd.DataFrame({

'Visitor_ID':[1,2,3],

'2010' :[4,3,10],

'2011' :[3,0,6],

'2012' :[6,7,3],

'2013' :[0,3,0]

})

df.set_index("Visitor_ID", inplace=True)

# Part 2, preparing the required variables.

def dictionary (max_visitors):

dictionary={}

for x in range(max_visitors):

dictionary["number_{}".format(x)] = []

# print(dictionary)

return dictionary

# Part 3, Figuring out if the customer returned.

def compare_yearly_visits(current_year, next_year):

index = 1

years = df.columns

for x in df[current_year]:

# print (df[years][current_year][index], 'this year.')

# print (df[years][next_year][index], 'Next year.')

how_many_visits = df[years][current_year][index]

did_he_return = df[years][next_year][index]

if did_he_return > 0:

# If the visitor returned, add to a bunch of formats:

returned_at_all.append([how_many_visits, True])

times_returned.append([how_many_visits, did_he_return])

dictionary["number_{}".format(x)].append(True)

else:

## If the visitor did not return, add to a bunch of formats:

returned_at_all.append([how_many_visits, False])

dictionary["number_{}".format(x)].append(False)

index = index +1

# Part 4, The actual program:

highest_amount_of_visits = 11 # should be done automatically, max(visits)?

relevant_years = len(df.columns) -1

times_returned = []

returned_at_all = []

dictionary = dictionary(highest_amount_of_visits)

for column in range(relevant_years):

# print (dictionary)

this_year = df.columns[column]

next_year = df.columns[column+1]

compare_yearly_visits(this_year, next_year)

print ("cumulative dictionary up to:", this_year,"\n", dictionary)