gpt4 book ai didi

python - 如何在 Python 中对两个数组求和?

转载 作者:行者123 更新时间:2023-11-28 20:21:19 24 4
gpt4 key购买 nike

我制作了一个这样的 DataFrame:

import numpy as np
import pandas as pd

occurrence = np.array([4, 5, 4, 0, 1, 4, 3])
year = np.array([1851,1852,1853,1854,1855,1856,1857])
disaster = {"occur":pd.Series(occur),"year":pd.Series(year)}
df = pd.DataFrame(disaster)

现在我想创建一个函数,当我给出两年时,它会给出这两年出现次数的总和。如果我输入 18511852,它会告诉我出现次数是 9

我写了一个这样的函数,但是它显示错误:

def dist(s1,s2):
return (sum (year>=s1 and year< s2))

print dist(s1,s2)

最佳答案

print(df.loc[df['year'].isin((1851,1852))]["occur"].sum())

或者:

 print(df.loc[df.year.isin((1851,1852))].occur.sum())

对于一系列日期,创建范围列表似乎比使用 & 更有效:

df.loc[df.year.isin(range(s1, s2+1))].occur.sum()

关于python - 如何在 Python 中对两个数组求和?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/28305302/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com