python - 如何使用Pasty创建dmatrix，而不必分别写出每个变量名？

假设我有一个很大的数据框，并且有一些列列表，并且我希望能够将它们放在小样的矩阵中，而不必单独写出每个名称。也就是说，我想从列表中将名称称为列名称列表以形成术语。而不是在数据框列中写出每一项。

例如下面的df

    df=pd.DataFrame( {'a':[1,2,3,4], 'b':[5,6,7,8],
                      'c':[8,4,5,3],  'd':[1,3,55,3],
                      'e':[8,4,5,3]})
   df
   >>
        a   b   c   d   e
   0    1   5   8   1   8
   1    2   6   4   3   4
   2    3   7   5   55  5
   3    4   8   3   3   3

据我了解，将其称为d矩阵需要我执行以下操作：

y,x = dmatrices('a~b+c+d+e', data=df)
但是，我希望能够按照以下方式运行更多功能：

   regress=['b', 'c']
   control=['e', 'd']
   y,x=dmatricies('a~{}+{}'.format(' '.join(e for e in regressors),
   ' '.join(c for c in control)), data=df)

但是，这是不成功的。

我还尝试使用包含两个条目的字典（例如，回归和控制），其中填充了列名称列表，然后将其输入到dmatricies的第一个条目中，但它也不起作用。

有没有人提出任何建议，以更有效的方式使事情进入patsy的困境，而不是写出我们希望包含在矩阵中的每个列名称？

在此先感谢您，让我知道是否不清楚。

最佳答案

在这里做for循环

for z in regress:
    for t in control:
        y,x=dmatrices('a~{}+{}'.format(z,t), data=df)
        print('a~{}+{}'.format(z,t))
        print(y,x)

a~b+e
[[1.]
 [2.]
 [3.]
 [4.]] [[1. 5. 8.]
 [1. 6. 4.]
 [1. 7. 5.]
 [1. 8. 3.]]
a~c+e
[[1.]
 [2.]
 [3.]
 [4.]] [[1. 8. 8.]
 [1. 4. 4.]
 [1. 5. 5.]
 [1. 3. 3.]]
a~d+e
[[1.]
 [2.]
 [3.]
 [4.]] [[ 1.  1.  8.]
 [ 1.  3.  4.]
 [ 1. 55.  5.]
 [ 1.  3.  3.]]

关于python - 如何使用Pasty创建dmatrix，而不必分别写出每个变量名？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/55886547/