如何将某变量取1的所有年份该变量都取1？

是 否 +2 论坛币

k人 参与回答

经管之家送您一份

应届毕业生专属福利!

求职就业群

赵安豆老师微信：zhaoandou666

经管之家联合CDA

送您一个全额奖学金名额~ !

立即领取

感谢您参与论坛问题回答

经管之家送您两个论坛币！

+2 论坛币

例如，year=2016时，某公司的一个变量inq=1，此时我想让该公司全部年份的inq都取1，或者生成一个新的变量treat，只要某一年这个公司的inq=1，该公司全部年份的treat=1，应该怎么操作呢？谢谢。

扫码加我 拉你入群

请注明：姓名-公司-职位

以便审核进群资格，未注明则拒绝

分享0 收藏1 回帖

关键词：treat 怎么操作 year Eat ear

by id:gen a=sum(inq)
by id:gen treat=(a!=0)

guanyifang50 发表于 2023-4-7 20:40
by id:gen a=sum(inq)
by id:gen treat=(a!=0)

谢谢！

您可以使用pandas库中的groupby和transform函数来实现这个任务。假设您的数据存储在一个名为df的DataFrame中，其中包含列'year', 'company_id' 和 'inq'。

首先，确保您的数据已经按照年份和公司ID排序：

```python
df = df.sort_values(['company_id', 'year'])
```

然后，您可以使用以下代码创建一个新的变量'treat'：

```python
df['treat'] = (df.groupby('company_id')['inq'].transform('max') == 1).astype(int)
```

这段代码的工作原理是首先按'company_id'分组，然后检查每个公司的'inq'的最大值是否为1。如果是，那么'treat'将被赋值为1；否则，赋值为0。

如果您只想在年份等于2016且'inq'为1的公司中设置所有年份的'inq'为1，可以稍微修改代码：

```python
df.loc[(df['year'] == 2016) & (df.groupby('company_id')['inq'].transform('max') == 1), 'inq'] = 1
```

这将只改变满足条件的行的'inq'值。

此文本由CAIE学术大模型生成，添加下方二维码，优先体验功能试用

by id:egen treat=sum(inq)
replace treat=1 if treat!=0