Создать серию панд

Question

Я хочу создать серию панд, которая содержит первые натуральные числа и соответствующие им квадраты. Первые цифры 'n' должны появляться в позиции индекса с помощью индексации вручную

Может кто-нибудь поделиться со мной кодом

Answer 1

Используйте numpy.arange с ** для квадратов:

n = 5
s = pd.Series(np.arange(n) ** 2)
print (s)
0     0
1     1
2     4
3     9
4    16
dtype: int32

Если хотите опустить 0:

n = 5
arr = np.arange(1, n + 1)
s = pd.Series(arr ** 2, index=arr)
print (s)
1     1
2     4
3     9
4    16
5    25
dtype: int32