Python-如何通过使用循环使此重复代码更短？

Question

我是Python的新手。 以下是我的数据示例：

Category    May  June  July
Product1    32   41    43
Product2    74   65    65
Product3    17   15    18
Product4    14   13    14

我有很多数据集，我想为每组计算卡方。 代码如下：

Product1 = [32,41,43]
chi2, p = scipy.stats.chisquare(Product1)
print('Product1')
if p > 0.05:
    print('Same')
else:
    print('Different')

Product2 = [74,65,65]
chi2, p = scipy.stats.chisquare(Product2)
print('Product2')
if p > 0.05:
    print('Same')
else:
    print('Different')

Product3 = [17,15,18]
chi2, p = scipy.stats.chisquare(Product3)
print('Product3')
if p > 0.05:
    print('Same')
else:
    print('Different')

Product4 = [14,13,14]
chi2, p = scipy.stats.chisquare(Product4)
print('Prokduct4')
if p > 0.05:
    print('Same')
else:
    print('Different')

我用“ df = pd.read_excel”插入数据表，它带有索引，我不知道如何调用每一行进行计算。

如何通过使用循环并从表中提取数据来缩短此重复代码？   非常感谢您的帮助。

Answer 1

您可以使用循环重复上述步骤，但是您最好利用scipy处理pandas数据帧的能力！您可以使用axis=1将chisquare测试应用于数据框的所有行。例如：

from scipy.stats import chisquare

df['p'] = chisquare(df[['May', 'June', 'July']], axis=1)[1]

df['same_diff'] = np.where(df['p'] > 0.05, 'same', 'different')

>>> df
   Category  May  June  July         p same_diff
0  Product1   32    41    43  0.411506      same
1  Product2   74    65    65  0.672294      same
2  Product3   17    15    18  0.869358      same
3  Product4   14    13    14  0.975905      same

现在您的数据框将您的p值作为一列，以及它们的值是“相同”还是“不同”作为列

Answer 2

将数据加载到熊猫数据框中后，我将开始：

然后，您可以执行以下操作：

for row in df.iterrows():
    product = row[1][0]
    chi, p = scipy.stats.chisquare(row[1][1:])
    print(product, ":", "same" if p > 0.05 else "different")

这将打印：

Product1 : same
Product2 : same
Product3 : same
Product4 : same