Python - Создание итогового фрейма данных из фрейма данных

Question

У меня есть такой фрейм данных,

DataFrame_A

Employee ID   A_ Status  C_Code  TestCol   Result_A  Result_B
20000         Yes        USA      asdasdq  True      False
20001         No         BRA      asdasdw  True      True
200002                   USA      asdasda  True      True 
200003        asda       MEX      asdasar  False     False

В этом фрейме данных Result_A и Result_B являются логическими столбцами.

Я хочу построить итоговый фрейм данныхчерез функцию, так что я могу повторно использовать.

Мне нужны следующие столбцы в моем фрейме данных, и вывод для Result_A выглядит как ниже, а Result_B другой логический столбец будет следующимстрока итогового фрейма данных.

 Name of the Column     No. of Records     No. of Employees    True_Records    False_Records     A_Status_Yes  A_Status_No     Mex_True      Mex_False      USA_True     USA_False
         Result_A              4               4                    3                     1                1            1               0            1              2              2  

Также следует отметить, что идентификатор сотрудника может иногда быть идентификатором EMPLOYEE или Employee_ID или EMPLOYEE_ID или EMPL_ID.Итак, список должен быть внутри python, и только один из них будет присутствовать внутри функции

. В реальном времени у меня есть 25 фреймов данных, поэтому я ищу функцию, которую я могу использовать и добавлять повторно.

Пожалуйста, помогите мне с этим.

Answer 1

Я думаю, что получил то, что вы хотите:

1- Создайте заново df:

df = pd.DataFrame({"Employee ID": [20000, 20001, 200002, 200003],
                  "A_ Status": ["Yes", "No", np.nan, "asda"],
                  "C_Code": ["USA", "BRA", "USA", "MEX"],
                  "TestCol": ["asdasdq", "asdasdw", "asdasda", "asdasar"],
                  "Result_A": [True, True, True, False],
                  "Result_B": [False, True, True, False]}, 
                  columns=["Employee ID", "A_ Status", "C_Code", "TestCol", "Result_A", "Result_B"])

2 - Создать второй фрейм данных df2:

df2 = pd.DataFrame(columns=["Name of the Column","No. of Records","No. of Employees","True_Records","False_Records","A_Status_Yes","A_Status_No","Mex_True","Mex_False","USA_True","USA_False"])

3- Результаты вычислений:

for column in df.columns[4:]: # For each columns of name pattern `Result_xx`
    print(column)
    a = [column,
        len(df["Employee ID"]), # Not sure about this one
        len(df["Employee ID"]),
        len(df[df[column] == True]),
        len(df[df[column] == False]),
        len(df[df["A_ Status"] == "Yes"]),
        len(df[df["A_ Status"] == "No"]),
        len(df[(df["C_Code"] == "MEX") & (df[column] == True)]),
        len(df[(df["C_Code"] == "MEX") & (df[column] == False)]),
        len(df[(df["C_Code"] == "USA") & (df[column] == True)]),
        len(df[(df["C_Code"] == "USA") & (df[column] == False)])
       ] # Create line as list

    df2.loc[len(df2), :] = a # Append line

4- Результаты:

+----+----------------------+------------------+--------------------+----------------+-----------------+----------------+---------------+------------+-------------+------------+-------------+
|    | Name of the Column   |   No. of Records |   No. of Employees |   True_Records |   False_Records |   A_Status_Yes |   A_Status_No |   Mex_True |   Mex_False |   USA_True |   USA_False |
|----+----------------------+------------------+--------------------+----------------+-----------------+----------------+---------------+------------+-------------+------------+-------------|
|  0 | Result_A             |                4 |                  4 |              3 |               1 |              1 |             1 |          0 |           1 |          2 |           0 |
|  1 | Result_B             |                4 |                  4 |              2 |               2 |              1 |             1 |          0 |           1 |          1 |           1 |
+----+----------------------+------------------+--------------------+----------------+-----------------+----------------+---------------+------------+-------------+------------+-------------+