咨询DID

请问大神们，针对既有试点城市名单又有试点企业名单的政策，如果我的样本数据是企业层面的，在做did时，可以以企业所在的试点城市为准来做吗，就是如果该企业在试点城市，那么treat赋值为1，不在则赋值为0。以试点企业为准来做的话，怕处理组太少了