为某些列循环所有值

Question

假设我的数据框架具有类似的结构，如下所示：

import pandas as pd
a_choise = ["True", "False", "False", "False", "True", "False", "False", "True", "True"]
b_choise = ["True", "True", "False", "False", "False", "False", "True", "True", "True"]
c_choise = ["False", "False", "True", "False", "True", "True", "False", "True", "False"]
a_n = ["a1", "a2", "a3", "a4", "a5", "a6", "a7", "a8", "a9"]
b_n = ["b1", "b2", "b3", "b4", "b5", "b6", "b7", "b8", "b9"]
c_n = ["c1", "c2", "c3", "c4", "c5", "c6", "c7", "c8", "c9"]
df = pd.DataFrame(
    {"a": list(range(1, 10)), "b": list(range(11, 20)), "c": range(21, 30), 
     "a_Cho":a_choise, "b_Cho":b_choise, "c_Cho":c_choise,
     "a_n":a_n, "b_n":b_n, "c_n":c_n}
)
    a   b   c   a_Cho   b_Cho   c_Cho   a_n b_n c_n
0   1   11  21  True    True    False   a1  b1  c1
1   2   12  22  False   True    False   a2  b2  c2
2   3   13  23  False   False   True    a3  b3  c3
3   4   14  24  False   False   False   a4  b4  c4
4   5   15  25  True    False   True    a5  b5  c5
5   6   16  26  False   False   True    a6  b6  c6
6   7   17  27  False   True    False   a7  b7  c7
7   8   18  28  True    True    True    a8  b8  c8
8   9   19  29  True    True    False   a9  b9  c9

我想要一个新的2列(Choise，Value)，满足"a_Cho“、"b_Cho”和"c_Cho“中所有值的折叠条件。

• 如果"a_Cho“= true然后choise = "a_n"，则值=a对应于"a_Cho”elif "a_Cho“= false的相应值，如果"b_Cho”= true然后choise = "b_n"，值=b对应的值"b_Cho"elif "b_Cho“= false，则移动到next
• ，如果"c_Cho”= true然后choise = "c_n"，值=c，对应的值为"c_Cho"elif "c_Cho“= false，然后移动到下一个
• ，如果"x_Cho”= false，而choise = "Invalide"

Answer 1

编辑过的

谢谢您对问题的更新。根据您的编辑，我相信.loc将对您有用。.loc允许我们执行“逻辑索引”，根据列等式抓取行。

例如，下面获取列a_Cho等于"True"的所有行，

>>> df.loc[df.a_Cho.eq('True'), ['a_n', 'a']]
    a   b   c   a_Cho   b_Cho   c_Cho   a_n b_n c_n
0   1   11  21  True    True    False   a1  b1  c1
4   5   15  25  True    False   True    a5  b5  c5
7   8   18  28  True    True    True    a8  b8  c8
8   9   19  29  True    True    False   a9  b9  c9

我们还可以使用.loc选择列的子集。

>>> df.loc[df.a_Cho.eq("True"), ["a_n", "a"]].rename(columns={"a_n": "choise", "a": "value"})

我用.rename()重命名了列。

使用上述方法，我们可以对您所述的每个条件执行逻辑索引，然后将结果连接起来。

def new_col_names(x):
    return {x + "_n": "choise", x: "value"}

# logical criteria
only_a = df.a_Cho.eq("True")
only_b = df.a_Cho.eq("False") & df.b_Cho.eq("True")
only_c = df.a_Cho.eq("False") & df.b_Cho.eq("False") & df.c_Cho.eq("True")
invalid = df.a_Cho.eq("False") & df.b_Cho.eq("False") & df.c_Cho.eq("False")

df_a = df.loc[only_a, ["a_n", "a"]].rename(columns=new_col_names("a"))
df_b = df.loc[only_b, ["b_n", "b"]].rename(columns=new_col_names("b"))
df_c = df.loc[only_c, ["c_n", "c"]].rename(columns=new_col_names("c"))
df_inv = df.loc[invalid].assign(choise="invalide", value="invalide").copy()
df_inv = df_inv[["choise", "value"]]

df_new = pd.concat([df_a, df_b, df_c, df_inv])

最后的DataFrame如下所示：

    choise    value
0   a1         1
4   a5         5
7   a8         8
8   a9         9
1   b2        12
6   b7        17
2   c3        23
5   c6        26
3   invalide  invalide

注意，左边的索引值显示了每个条目的原始行号。如果您不关心这些数字，可以将ignore_index=True选项传递给.concat。

这离你想要的更近吗？

原创

你好，欢迎来到StackOverflow！我不确定我是否完全理解你的问题。例如，在示例代码中，似乎没有在每次迭代中使用循环变量x。看到与您正在使用的示例DataFrame具有相同结构的示例可能会有所帮助。

我的印象是，您的问题可能与使用this question的pd.melt类似。

您的DataFrame结构是否如下所示？

>>> import pandas as pd
>>> df = pd.DataFrame(
    {"A": list(range(1, 10)), "B": list(range(11, 20)), "C": range(21, 30)}
)
>>> df.head()

    A   B   C
0   1   11  21
1   2   12  22
2   3   13  23
3   4   14  24
4   5   15  25
5   6   16  26
6   7   17  27
7   8   18  28
8   9   19  29

如果是这样，您可以使用pd.melt对其进行重构，使其具有两个列，即“列名”列和“列值”列，正如您在问题中所描述的那样。

上面示例DataFrame的命令和输出是：

>>> pd.melt(df, value_vars=['A', 'B', 'C'])

  variable  value
0   A   1
1   A   2
2   A   3
3   A   4
4   A   5
5   A   6
6   A   7
7   A   8
8   A   9
9   B   11
10  B   12
11  B   13
12  B   14
13  B   15
14  B   16
15  B   17
16  B   18
17  B   19
18  C   21
19  C   22
20  C   23
21  C   24
22  C   25
23  C   26
24  C   27
25  C   28
26  C   29

这和你要求的类似吗？如果不是，您是否可以提供您正在使用的DataFrame的示例以及您希望最终结果看起来如何的示例？它可以是一个简化的或“模拟”的例子。