我有两个numpy矩阵A和B两者都是数字矩阵，仅由条目0或1组成.
我想从这两个矩阵计算矩阵积C，该矩阵在A(行、列)方面遵守以下规则:
If A(i,j) = 1 and B(i,j)=1, C(i,j)=1.
If A(i,j) = 1 and B(i,j)=0, C(i,j)=-q.
If A(i,j) = 0, C(i,j) = 0, regardless of the value of B(i,j).
其中q满足方程p-(n-p)*q=0，其中p是B(i，:)中1的数量.
我可以用几个for循环来实现这个逻辑，但我很想问是否有一种单行Python方法来实现这个逻辑.
IIUC您可以做:
假设您有两个矩阵5x 2:
# A
[[0 1 0 1 1]
[1 1 1 1 0]]
# B
[[0 0 1 1 1]
[0 0 1 1 0]]
然后:
m, n = A.shape # 5, 2
p = B.sum(axis=1) # [3 2]
q = p / (n - p) # [1.5 0.66666667]
m1 = A & B
m2 = (A & ~B) * -q[:, None]
print(m1 + m2)
输出:
[[ 0. -1.5 0. 1. 1. ]
[-0.66666667 -0.66666667 1. 1. 0. ]]