在 for 循環中實例化 python dataframe

Question

我正在嘗試使用 for 循環和 if 語句為 python dataframe object 創建條件。為了准確指定在特定條件下從 data.table 中提取哪一行，我搜索了行索引，並創建了一個參數來指定位置在 for 循環之前。 具體看起來像這樣：

import pandas as pd

input_csv_file = "./CSV/Officers_and_Shareholders.csv"

df = pd.read_csv(input_csv_file, skiprows=10, on_bad_lines='skip')
df.fillna('', inplace=True)
# df.drop([0, 3], inplace=True)
df.columns = ['Nama', 'Jabatan', 'Alamat', 'Klasifikasi Saham', 'Jumlah Lembar Saham', 'Total']

# print(df.shape)
# print(df.columns)
# print(df.iloc[:53])

# shareholders = df.iloc[24:42]
# print(shareholders)

# officers = df.iloc[0:23]
# print(officers)

dataframe = df.query("Total.ne('-')")

def get_shareholder_by_row_index():
    for column in df.columns:
        if object(df.iloc[column][:53]) == dataframe:
            shareholders = df.iloc[24:42]
            print(shareholders)
        # elif object(df[:53][column]) != dataframe:
        #     officers = df.iloc[0:23]
        #     print(officers)

因為CSV文件的格式不對，我強制dataframe在原來的CSV文件之上重新創建了一個header，我在df.columns下標明。 df.iloc[24:42]和df.iloc[0:23]能夠具體定位到 dataframe 中的數據范圍，但是在 for 循環中實例化時並沒有返回。 客觀地，我想創建一個 function ，如果 Total 列下的行為空（-），則 return the officers ，但如果 Total 列下的行不為空，則 return shareholders 。 在這種情況下，我應該如何修改for循環和if語句呢？

股東所需的output將是：

24       PT CTCORP INFRASTRUKTUR D INDONESIA,  ...  Rp. 3.200.000.000
25                              Nomor SK :- I  ...                   
26      JalanKaptenPierreTendeanKavling12-14A  ...                   
27              PT INTRERPORT PATIMBAN AGUNG,  ...  Rp. 2.900.000.000
28                                Nomor SK :-  ...                   
29                                             ...                   
30                                             ...                   
31                                             ...                   
32                                             ...                   
33                                             ...                   
34                  PT PATIMBAN MAJU BERSAMA,  ...  Rp. 2.900.000.000
35                             Nomor SK :AHU-  ...                   
36                0061318.AH.01.01.TAHUN 2021  ...                   
37              Tanggal SK :30 September 2021  ...                   
38                                             ...                   
39                                             ...                   
40                      PT TERMINAL PETIKEMAS  ...  Rp. 1.000.000.000
41                                  SURABAYA,  ...                   
42                                Nomor SK :-  ... 

對於官員，它將返回：

                                         Nama  ...              Total
1                       NIK: 3171060201830005  ...                   
2                       NPWP: 246383541071000  ...                   
3               TTL: Jakarta, 02 Januari 1983  ...                   
5                       NIK: 1271121011700003  ...                   
6                       NPWP: 070970173112000  ...                   
7                TTL: Bogor, 10 November 1970  ...                   
8                         ARLAN SEPTIA ANANDA  ...                   
9                                      RASAM,  ...                   
10                      NIK: 3174051209620003  ...                   
11                      NPWP: 080878200013000  ...                   
12                 TTL: Jakarta, 12 September  ...                   
13                                       1962  ...                   
15                      NIK: 3171011605660004  ...                   
16                      NPWP: 070141650093000  ...                   
17                  TTL: Jakarta, 16 Mei 1966  ...                   
18                                FUAD RIZAL,  ...                   
21  PURNOMO, UTAMA RASRINIK: 3578032408610001  ...                   
22                      NPWP: 097468813615000  ...                   
23               TTL: SLEMAN, 24 Agustus 1961  ... 

              

Answer 1

利益相關者和官員將被打印到索引（行號），如果這不是所需的答案，那么請提及一些細節

def get_shareholder_by_row_index():
    for i in range(len(df)): 
# this will give you shareholders if row under Total is empty else office if row is not empty 
        if df["Total"][i] == '' :
            print(i," shareholders")
            print(df.iloc[i])
            # what ever your code is, will be here
        else:
            print(i," officers")
            print(df.iloc[i])

            # what ever your code is, will be here

# this will give you the indces where row under total is empty
    print(df["Total"].iloc[:53][df["Total"] == ''])