2017-10-17 86 views
1

我正在爲我的編碼類在文檔字符串中列出的這個問題工作。我將不勝感激任何關於優化我的代碼的幫助,以及爲什麼儘管重置索引時仍然收到以下錯誤的任何解釋。使用過濾器對大熊貓進行查詢和排序,導致未解決的錯誤

import pandas as pd 
def beds_top_ten(df, facility_id): 
    ''' 
    INPUT: DataFrame, int 
    OUTPUT: date 
    Write a pandas query that returns the ten census dates with the highest 
    number of available beds for the nursing home with the specified facility id 
    REQUIREMENTS: 
    Do a filter followed by a sort rather than a sort followed by a merge. 
    ''' 
    df = pd.read_csv('beds.csv', low_memory= False) 
    df['Bed Census Date'] = pd.to_datetime(df['Bed Census Date']) 
    df = df.filter(items =['Facility ID', 'Bed Census Date','Available Residential Beds']) 
    df = df.sort_values(by =[ 'Facility ID', 'Available Residential Beds'], ascending= False) 
    df_group_by_ten = df.groupby('Facility ID').head(10).reset_index(drop=True) 
    dates = df_group_by_ten.loc[df_group_by_ten['Facility ID']==facility_id, 'Bed Census Date'] 
    return dates 

這是什麼表看起來像第一GROUPBY後:

Facility ID Bed Census Date Available Residential Beds 
336 19 2011-01-05 29 
339 19 2010-12-15 28 
330 19 2011-02-23 27 
332 19 2011-02-02 27 
333 19 2011-01-26 27 
334 19 2011-01-19 27 
335 19 2011-01-12 27 
338 19 2010-12-22 27 
341 19 2010-12-01 27 
331 19 2011-02-09 26 
16 17 2013-04-10 22 
87 17 2011-11-09 19 
30 17 2013-01-02 17 
37 17 2012-11-07 17 
47 17 2012-08-29 17 
31 17 2012-12-26 16 
56 17 2012-06-20 16 
10 17 2013-05-22 15 
27 17 2013-01-23 15 
61 17 2012-05-16 15 

當我從我的COMMAND_LINE運行:

In [15]: beds_top_ten('beds.csv',17) 
Out[15]: 
16 2013-04-10 
87 2011-11-09 
30 2013-01-02 
37 2012-11-07 
47 2012-08-29 
31 2012-12-26 
56 2012-06-20 
10 2013-05-22 
27 2013-01-23 
61 2012-05-16 
Name: Bed Census Date, dtype: datetime64[ns] 

然而,當我運行在相同的代碼在線環境中,我收到以下錯誤:

/usr/local/lib/python2.7/unittest/suite.py:108: DtypeWarning: Columns (10,45) have mixed types. Specify dtype option on import or set low_memory=False. 
    test(result) 
E 
====================================================================== 
ERROR: test_fourth_pandas (test_methods.Test) 
---------------------------------------------------------------------- 
Traceback (most recent call last): 
    File "/usr/src/app/test_methods.py", line 25, in test_fourth_pandas 
    all_equal = np.all(result == answer) 
    File "/usr/local/lib/python2.7/site-packages/pandas/core/ops.py", line 812, in wrapper 
    raise ValueError(msg) 
ValueError: Can only compare identically-labeled Series objects 

---------------------------------------------------------------------- 
Ran 1 test in 19.743s 

FAILED (errors=1) 
+0

我覺得你這得太多。這應該是足夠的:'df [df ['Facility ID'] == facility_id] .sort_values('Available Residential Beds',ascending = False).head(10)' –

+0

@COLDSPEED,謝謝,這條線確實有助於簡化代碼,但我仍然收到相同的錯誤。 – whd

+0

請參閱下面的答案。 –

回答

1

pd.to_datetime沒有問題。你可能有錯誤的日期。嘗試指定一種格式,並將errors='coerce這樣無效的格式轉換爲NaT。現在

df['Bed Census Date'] = pd.to_datetime(df['Bed Census Date'].str.strip(), 
          format='%Y-%m-%d', errors='coerce') 

,擴大我的comment,篩選,排序,並得到使用head第10項:

x = df[df['Facility ID'] == facility_id]\ 
     .sort_values('Available Residential Beds', ascending=False).head(10) 
return x['Bed Census Date'] 
0

刪除日期格式化行解決了上述錯誤。

df = pd.read_csv('beds.csv', low_memory= False) 
    #df['Bed Census Date'] = pd.to_datetime(df['Bed Census Date']) 
    df = df.filter(items=['Facility ID', 'Bed Census Date','Available Residential Beds']) 
    x = df[df['Facility ID'] == facility_id].sort_values('Available Residential Beds', ascending=False).head(10) 
    return x['Bed Census Date'] 
+0

看來你用我的答案來解決你的問題,所以你可以選擇[接受](https://stackoverflow.com/help/someone-answers)我的[answer](https://stackoverflow.com/a/ 46798352/4909087)或在您的信貸礦。謝謝。 –

+0

非常感謝!我已經這樣做了。 – whd