add sd.py

This commit is contained in:
chenwj113 2022-10-12 16:54:26 +08:00
parent 9073e118e3
commit f937c697a4
1 changed files with 16 additions and 0 deletions

16
sd.py Normal file
View File

@ -0,0 +1,16 @@
import numpy as np
import pandas as pd
a = np.arange(1000)
l = set([ f"{i:03d}" for i in a ])
sum_dict = dict()
print(len(l))
for i in l:
_sum = sum(map(int, i))
v = sum_dict.get(_sum, 0)
sum_dict[_sum] = v + 1
df = pd.Series(sum_dict)
print(df.sort_values(ascending=False))