擅长:python、mysql、java
<p>你可以得到你的过滤组的元素平均值,方式如下。一些中间步骤是必要的(重塑数据并将列表转换为numpy数组),但是这些步骤应该产生您想要的方法列表(或数组)。你知道吗</p>
<pre><code># melt H1 and H2 columns into key-value columns
# this will make it easier to select either the H1 or H2 list
df = pd.melt(df, id_vars=['Location', 'Set', 'Dir'], \
value_vars=['H1', 'H2'], var_name="Target_Dir", value_name="Values")
# convert lists to numpy arrays
# in order to be able to specify the axis for the mean calculation
df.Values = df.Values.apply(np.array)
# filter df to your target Dirs, group by Set
# and calculate element-wise means
df[df['Dir'] == df['Target_Dir']].groupby('Set')['Values'].apply(lambda x: np.mean(x, axis=0))
</code></pre>