Question

我创建了一个类对象，它从数据库中检索信息并将其存储在pandas中，这样我就可以使用一些数据科学库来进行操作。

class IntDailyGoals (object):
    def __init__(self, begin_date, end_date, store=None):
        self.begin_date = begin_date
        self.end_date = end_date
        self.store = store
        self.int_mnth_goal = pd.DataFrame(list(StoreGoalsInput.objects.values('store_number',
                                                                              'interest',
                                                                              'date')))
        self.int_mnth_goal['interest'] = pd.to_numeric(self.int_mnth_goal['interest'])
        self.int_mnth_goal['date'] = pd.to_datetime(self.int_mnth_goal['date'])
        self.mnth_goal_int =self.int_mnth_goal[(self.int_mnth_goal['date'] >= self.begin_date) &
                                               (self.int_mnth_goal['date'] <= self.end_date) &
                                               (self.int_mnth_goal['store_number'] == self.store.store_number)]
        self.mnth_goal_int= self.mnth_goal_int['interest']
        self.tot_workingdays = np.busday_count(np.datetime64(self.begin_date),
                                               np.datetime64(self.end_date),
                                               weekmask='Mon Tue Wed Thu Fri Sat')
        self.div_intmnthgoal_workingdays = round(np.divide(self.mnth_goal_int, self.tot_workingdays),2)


    def get_div_goalsint_wdays(self):
        div_goalsint_wdays = self.div_intmnthgoal_workingdays
        return div_goalsint_wdays

    def __str__(self):
        return self.get_div_goalsint_wdays()

返回：

2    6558.4 
Name: interest, dtype: float64

我只需要它返回整数6558.4，因为这是在Django模板中显示的。

我试过这个：

    def get_div_goalsint_wdays(self):
    div_goalsint_wdays = self.div_intmnthgoal_workingdays['interest']
    return div_goalsint_wdays

但我得到KeyError: 'interest'

Answer 1

您描述的结果是具有单一值的熊猫系列要获取它包含的浮点值，您应该使用：

self.div_intmnthgoal_workingdays.iloc[0]

Answer 2

您可以使用.values属性或.tolist方法获取值（不带索引）。

例如，如果我有Series：

>>> s
2    3
dtype: int64

然后值是：

>>> s.values
array([3], dtype=int64)
>>> s.tolist()
[3]

这可以像任何其他类似序列的类一样编入索引：

>>> s.values[0]
3
>>> s.tolist()[0]
3

使用带有dtype：float64的np.divide函数时，只返回数字

2 个答案: