Умножьте DataFrame на другую форму DataFrame (или серии) - PullRequest
0 голосов
/ 12 октября 2018

У меня есть этот DataFrame, подобный этому:

1  2  1  3  1  4
2  4  5  1  1  4
1  3  5  3  1  4
1  3  1  3  1  4

Другой, подобный этому

1  1  0  0  0  0

Я хочу умножить их так, чтобы я получил

1  2  0  0  0  0
2  4  0  0  0  0
1  3  0  0  0  0
1  3  0  0  0  0

Так что происходит то, что каждый столбец с 1 во втором df умножает каждое значение на один, а каждый столбец с нулем меняет все столбцы в первом кадре данных на 0.

Ответы [ 2 ]

0 голосов
/ 12 октября 2018

Использование pandas mul

df1.mul(df2.T[0])
Out[22]: 
   1  2  3  4  5  6
0  1  2  0  0  0  0
1  2  4  0  0  0  0
2  1  3  0  0  0  0
3  1  3  0  0  0  0
0 голосов
/ 12 октября 2018

Вероятно, проще всего использовать лежащие в основе массивы, и пусть numpy сделает магию вещания:

>>> df1.values * df2.values
array([[1, 2, 0, 0, 0, 0],
       [2, 4, 0, 0, 0, 0],
       [1, 3, 0, 0, 0, 0],
       [1, 3, 0, 0, 0, 0]])

Вы можете поместить его обратно в кадр данных с теми же столбцами, что и df1, используя:

>>> pd.DataFrame(df1.values * df2.values, columns=df1.columns)
   0  1  2  3  4  5
0  1  2  0  0  0  0
1  2  4  0  0  0  0
2  1  3  0  0  0  0
3  1  3  0  0  0  0

Или, если вы не против перезаписи df1:

>>> df1[:] = df1.values * df2.values
>>> df1
   0  1  2  3  4  5
0  1  2  0  0  0  0
1  2  4  0  0  0  0
2  1  3  0  0  0  0
3  1  3  0  0  0  0

В качестве альтернативы, вы можете повеселиться с np.broadcast_to, если хотите:

>>> df1*np.broadcast_to(df2,df1.shape)
   0  1  2  3  4  5
0  1  2  0  0  0  0
1  2  4  0  0  0  0
2  1  3  0  0  0  0
3  1  3  0  0  0  0
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...