Question

我正在编写一个函数，该函数返回一个字典，其中数据集中所有引用的创建日期的年份用作键，并作为值，它指定由函数{{1}返回的两个项的元组}。

do_get_citations_per_year

我不断收到错误消息：

def do_get_citations_per_year(data, year):
    result = tuple()
    my_ocan['creation'] = pd.DatetimeIndex(my_ocan['creation']).year

    len_citations = len(my_ocan.loc[my_ocan["creation"] == year, "creation"])
    timespan = my_ocan.loc[my_ocan["creation"] == year, "timespan"].fillna(0).mean()

    result = (len_citations, round(timespan))

    return result

def do_get_citations_all_years(data):
    mydict = {}
    s = set(my_ocan.creation)
    print(s)
    for year in s:
        mydict[year] = do_get_citations_per_year(data, year)
    #print(mydict)
    return mydict

更新：为了提供一个有效的示例，我在这里发布了另一个函数，特别是处理我的数据帧（my_ocan）(32, 240) {2016, 2017, 2018, 2013, 2015} File "/Users/lisa/Desktop/yopy/execution_example.py", line 28, in <module> print(my_ocan.get_citations_all_years()) File "/Users/lisa/Desktop/yopy/ocan.py", line 35, in get_citations_all_years return do_get_citations_all_years(self.data) File "/Users/lisa/Desktop/yopy/lisa.py", line 113, in do_get_citations_all_years mydict[year] = do_get_citations_per_year(data, year) File "/Users/lisa/Desktop/yopy/lisa.py", line 103, in do_get_citations_per_year result = (len_citations, round(timespan)) ValueError: cannot convert float NaN to integer Process finished with exit code 1和解析函数do_process_citation_data(f_path)的函数：

parse_timespan

当我打印my_ocan ['timespan']

我得到：

def do_process_citation_data(f_path):
    global my_ocan

    my_ocan = pd.read_csv(f_path, names=['oci', 'citing', 'cited', 'creation', 'timespan', 'journal_sc', 'author_sc'],
                          parse_dates=['creation', 'timespan'])
    my_ocan = my_ocan.iloc[1:]  # to remove the first row
    my_ocan['creation'] = pd.to_datetime(my_ocan['creation'], format="%Y-%m-%d", yearfirst=True)
    my_ocan['timespan'] = my_ocan['timespan'].map(parse_timespan)

    print(my_ocan['timespan'])

    return my_ocan

    #print(my_ocan['timespan'])

timespan_regex = re.compile(r'P(?:(\d+)Y)?(?:(\d+)M)?(?:(\d+)D)?')
def parse_timespan(timespan):
    # check if the input is a valid timespan
    if not timespan or 'P' not in timespan:
        return None

    # check if timespan is negative and skip initial 'P' literal
    curr_idx = 0
    is_negative = timespan.startswith('-')
    if is_negative:
        curr_idx = 1

    # extract years, months and days with the regex
    match = timespan_regex.match(timespan[curr_idx:])

    years = int(match.group(1) or 0)
    months = int(match.group(2) or 0)
    days = int(match.group(3) or 0)

    timespan_days = years * 365 + months * 30 + days

    return timespan_days if not is_negative else -timespan_days

我认为问题是0.0

我该如何解决将NaN转换为整数的问题？

提前谢谢！

Answer 1

我已尝试使用python 2.7：

>>> round(float('NaN'))
nan
>>> round(float(0.0))
0.0

而在python 3.6中是这样的：

>>> round(float('NaN'))
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: cannot convert float NaN to integer
>>> round(float(0.0))
0

因此，您似乎将任何NaN值都加入了round函数。您可以使用try try语句来解决此问题：

try:
    result = (len_citations, round(timespan))
except ValueError:
    result = (len_citations, 0)

解决ValueError：无法将float NaN转换为整数

1 个答案: