在原地对数据帧进行切片,并在循环中动态重命名



我知道这可能不是一个好的做法,但我很想知道是否可以取两个dfs(在本例中为srm和srae(,取每个dfs的一个切片,然后将这个切片的数据帧重命名为srm1和srae1。

逻辑如下。

for x in (srm, srae):
x1 = x[x['years_in_role']>5]
print(x.shape, x1.shape)

您可以将2个元组解压缩为2个变量:

srm1, srae1 = [x[x['years_in_role']>5] for x in (srm, srae)]

您的解决方案应该用于创建列表,然后创建新的变量:

L = []
for x in (srm, srae):
x1 = x[x['years_in_role']>5]
L.append(x1)
srm1 = L[0]
srae1 = L[1]

最新更新