py_scripts/pd_test.py

21 lines
467 B
Python

import pandas as pd
import numpy as np
s1 = pd.Series([1.47, 2.3, 3.0])
s2 = pd.Series([5.5, 6.1])
s3 = pd.Series([2.3, 4.6, 5.3])
s1.index = [True, False, False]
s2.index = [False, True]
s3.index = [False, True, False]
a1 = np.expand_dims(s1.to_numpy(), axis=0)
a2 = np.expand_dims(s2.to_numpy(), axis=0)
a3 = np.expand_dims(s3.to_numpy(), axis=0)
print(a1.shape)
print(a2.T.shape)
print((a1.T)*a2)
t = (a2.T)*a1
print(len(t.flatten()))
print((s1*s2).loc[True])