Блог

Допустим, у нас есть эти данные:

 list1, list2, list3 = [1,2,3,4], [1990, 1990, 1990, 1991], [2009, 2009, 2009, 2009]
df = pd.DataFrame(list(zip(list1, list2, list3)), columns = ['Index', 'Y0', 'Y1'])

> df

Index  Y0          Y1
1      1990        2009
2      1990        2009
3      1990        2009
4      1991        2009
  

Я хочу подсчитать за каждый год, сколько строк («индекс») приходится на каждый год, но исключая Y0.

Допустим, мы начнем с первого доступного года, 1990:

Сколько строк мы насчитываем? 0.

1991:

• Три (строка 1, 2, 3)

1992:

• Четыре (строка 1, 2, 3, 4)

…

2009:

• Четыре (строка 1, 2, 3, 4)

Итак, я хочу получить фрейм данных, в котором говорится:

 Count  Year
0      1990     
3      1991     
4.     1992
...    ...    
4      2009     
  

Моя попытка:

 df['Y0'] = pd.to_datetime(df['Y0'], format='%Y')
df['Y1'] = pd.to_datetime(df['Y1'], format='%Y')

# Group by the interval between Y0 and Y1 
df = d.groupby([d['Y0'].dt.year, d['Y1'].dt.year]).agg({'count'})
df.columns = ['count', 'Y0 count', 'Y1 count']

# sum the total
df_sum = pd.DataFrame(df.groupby(df.index)['count'].sum())
  

Но результат не выглядит правильным.

Ценю любую помощь.

вы могли бы сделать:

 min_year = df[['Y0', 'Y1']].values.min()
max_year = df[['Y0', 'Y1']].values.max()
year_range = np.arange(min_year, max_year 1)
counts = ((df[['Y0']].values < year_range) amp; (year_range<= df[['Y1']].values)).sum(axis=0)
o = pd.DataFrame({"counts": counts, 'year': year_range})
  

 counts  year
0   0   1990
1   3   1991
2   4   1992
3   4   1993
4   4   1994
5   4   1995
6   4   1996
7   4   1997
8   4   1998
9   4   1999
10  4   2000
11  4   2001
12  4   2002
13  4   2003
14  4   2004
15  4   2005
16  4   2006
17  4   2007
18  4   2008
19  4   2009
  

Следующее должно выполнить вашу работу:

 counts=[]
years=[]

def count_in_interval(year):
    n=0
    for i in range(len(df)):
        if df['Y0'][i]<year<=df['Y1'][i]:
            n =1
    return n

for i in range(1990, 2010):
    counts.append(count_in_interval(i))
    years.append(i)

result=pd.DataFrame(zip(counts, years), columns=['Count', 'Year'])