python - Pandas 过滤和组合

Question

我是 pandas 新手，很难弄清楚如何执行以下操作

data = pd.read_csv('data.csv', sep=';', header=0, names=['a1', 'a2', 'b1', 'b2', 'c1', 'c2'])

所有值都是整数

我需要过滤掉“a1”、“b1”、“c1”小于 5 的行，然后生成一个仅包含“a”、“b”、“c”的新 DataFrame，其中

a[i] = a2[i] / a1[i]
c[i] = b2[i] / b1[i]
d[i] = c2[i] / c1[i]

Answer 1

要过滤掉，您可以使用:

data = data[(data.a1 < 5) & (data.b1 < 5) & (data.c1 < 5)]

然后您可以创建一个包含新列的 DataFrame:

pd.DataFrame({
    'a': data.a2 / data.a1,
    'c': data.b2 / data.b1,
    'd': data.c2 / data.c1})

(请注意，在前面的内容中，您可能会考虑使用

    'a': data.a2.astype(float) / data.a1,

(等等)，因为整数除法可能不是您想要的。)