Несколько итераций над пандой DataFrame - PullRequest
0 голосов
/ 18 декабря 2018

Для каждого сайта и месяца я пытаюсь вычислить сумму значений на близлежащих сайтах, используя Pandas DataFrame и три цикла.Начальный DataFrame - df1, ниже.

| Site name | Lat   | Lon    | March 2018 | April 2018 | May 2018 |
|-----------|-------|--------|------------|------------|----------|
| A         | 10.0  | 15.0   | 1          | 2          | 3        |
| B         | 10.1  | 15.0   | 1          | 2          | 3        |
| C         | 12.0  | 100.0  | 1          | 2          | 3        |

Цикл 1) Начать с первого столбца, содержащего данные
Цикл 2) Захватить координаты первого сайта для сравнения со всеми другими сайтами
Цикл 3) Сравнитькоординаты первого сайта с каждого сайта;если расстояние между ними (рассчитанное по формуле Хаверсайна, не указано в списке) меньше Rtarget , добавьте к msum , который является месячной суммой значений для всех сайтов в пределах Rtarget расстояние от исходного сайта.

Выходные данные отправляются в пустой DataFrame, df2, с идентичным индексом и столбцами, как df1.На каждой итерации цикла 2 должно быть конечное значение мсум , которое должно быть введено в то же место (индекс и столбец) в df2, что и в цикле 2.

Мой код в настоящее время производит df2, который выглядит следующим образом:

            | Lat | Lon | March 2018 | April 2018 | May 2018 |
| Site name |-----|-----|------------|------------|----------|
| A         | 0   | 0   | 0          | 0          | 0        |
| B         | 0   | 0   | 0          | 0          | 0        |
| C         | 0   | 0   | 3          | 6          | 9        |

Хотя df2 должен выглядеть следующим образом:

            | Lat | Lon | March 2018 | April 2018 | May 2018 |
| Site name |-----|-----|------------|------------|----------|
| A         | 0   | 0   | 2          | 4          | 6        |
| B         | 0   | 0   | 2          | 4          | 6        |
| C         | 0   | 0   | 1          | 2          | 3        |

Мой код ниже:

# Create empty data frame with identical columns
df1 = pd.DataFrame(0, index=df.index, columns=df.columns)

Rtarget = 500  # Set the target distance to X miles

# Iterate through each month of data
for column, row in df.iloc[:,2:].iteritems():

    # Select the initial site and define the starting coordinates
    for index, row in df.iterrows():
        lat01 = row['Latitude']
        lon01 = row['Longitude']
        msum = 0 # set monthly sum equal to zero

        # Iterate through each comparison site and define comparison coordinates
        for index, row in df.iterrows():
            lat02 = row['Latitude']
            lon02 = row['Longitude']
            distance = haversine(lon01, lat01, lon02, lat02) 
            if distance <= Rtarget: 
                msum += row[column] # Add the value to the monthly sum

    # Place monthly sum into the corresponding location in the empty dataframe, df1
    df1[column][index] = msum

Что мне не хватает, из-за чего все строки, кроме последней, остаются пустыми?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...