Question

我正在尝试使用日期范围内的列表理解来创建列表。我有一个具有存储容量的数据框，当为某家公司建造存储时，我说我想获取每个日期的所有仓库的总容量。我知道如何使用常规的for循环并附加到列表中来做到这一点，但是我不知道如何使它与列表理解一起工作。

如下面的代码所示，在熊猫系列中尝试使用.sum()函数时遇到错误。我不知道这是否与数据帧切片有关，或者与列表代码本身所缺少的内容有关，我是否需要对条件切片？

import pandas as pd
from datetime import datetime
import numpy as np

#My date range
date_rng = pd.date_range(start = '01/01/2000', end = '31/12/2019')

#The dataframe warehouses contains a column with the operation start date and also the storage capacity of the warehouse
installed capacity = [warehouses[warehouses['StartDate']<=date]['Storage'].sum() for date in date_rng]
``´

 installed capacity = [warehouses[warehouses['StartDate']<=date]['Storage'].sum() for date in date_rng]
                     ^
SyntaxError: invalid syntax

Note: The invalid syntax is pointing to the '.sum()'

Answer 1

变量名中不允许使用空格，因此我们使用下划线代替空格。将变量已安装容量替换为installed_capacity

列表理解结合数据框求和功能

1 个答案: