Pandas: L oop для определенных c значений столбца

Question

Это очень короткий след. Но исходный файл слишком велик

highest_layer,transport_layer,src_ip,dst_ip,src_port,dst_port,ip_flag,packet_length,transport_flag,time,timestamp,geo_country,data
DNS,UDP,192.168.1.6,172.217.12.131,32631,53,0,89,-1,2020-06-10 19:38:08.863846,1591832288863,Unknown,
DNS,UDP,192.168.1.6,192.168.1.1,31708,53,0,79,-1,2020-06-10 19:38:08.864186,1591832288864,Unknown,
DNS,UDP,192.168.1.6,172.217.12.131,32631,53,0,79,-1,2020-06-10 19:38:08.866492,1591832288866,Unknown,
SSDP,UDP,192.168.1.6,172.217.12.131,32631,1900,0,216,-1,2020-06-10 19:38:08.887298,1591832288887,Unknown,
DNS,UDP,192.168.1.1,192.168.1.6,53,32631,16384,105,-1,2020-06-10 19:38:08.888232,1591832288888,Unknown,
TCP,TCP,192.168.1.6,172.217.12.131,32631,443,16384,78,2,2020-06-10 19:38:08.888553,1591832288888,Unknown,
DNS,UDP,192.168.1.1,192.168.1.6,53,31708,16384,95,-1,2020-06-10 19:38:08.895148,1591832288895,Unknown,
TCP,TCP,192.168.1.6,172.217.12.131,16807,443,16384,78,2,2020-06-10 19:38:08.895594,1591832288895,Unknown,
DNS,UDP,192.168.1.1,192.168.1.6,53,16807,16384,119,-1,2020-06-10 19:38:08.896202,1591832288896,Unknown,
TCP,TCP,192.168.1.6,172.217.12.131,16807,443,16384,78,2,2020-06-10 19:38:08.896540,1591832288896,Unknown,
DNS,UDP,192.168.1.6,172.217.12.131,16807,53,0,75,-1,2020-06-10 19:38:08.911968,1591832288911,Unknown,
DATA,UDP,192.168.1.3,192.168.1.6,51216,58185,16384,558,-1,2020-06-10 19:38:08.913276,1591832288913,Unknown,
TCP,TCP,172.217.12.131,192.168.1.6,443,53717,0,74,18,2020-06-10 19:38:08.916735,1591832288916,Unknown,
TCP,TCP,192.168.1.6,172.217.12.131,58185,443,16384,66,16,2020-06-10 19:38:08.916860,1591832288916,Unknown,
TLS,TCP,192.168.1.6,172.217.12.131,58185,443,16384,583,24,2020-06-10 19:38:08.917442,1591832288917,Unknown,
TCP,TCP,172.217.10.237,192.168.1.6,443,53718,0,74,18,2020-06-10 19:38:08.919293,1591832288919,Unknown,
TCP,TCP,192.168.1.6,172.217.12.131,58185,443,16384,66,16,2020-06-10 19:38:08.919423,1591832288919,Unknown,
TLS,TCP,192.168.1.6,172.217.12.131,32631,443,16384,583,24,2020-06-10 19:38:08.919593,1591832288919,Unknown,
TCP,TCP,172.217.11.14,192.168.1.6,443,53719,0,74,18,2020-06-10 19:38:08.928819,1591832288928,Unknown,
TCP,TCP,192.168.1.6,172.217.12.131,16807,443,16384,66,16,2020-06-10 19:38:08.928922,1591832288928,Unknown,
TLS,TCP,192.168.1.6,172.217.12.131,58185,443,16384,583,24,2020-06-10 19:38:08.929100,1591832288929,Unknown,

Я удалил несколько ненужных столбцов, и я хочу получить совокупную длину пакета из указанных c src_ip (192.168.1.6), IP-адреса назначения (172.217.12.131) и src_port (32631,16807, 58185). Я хочу перебрать src_port для заданных src_ip и dest_ip. В этом случае для каждого из 3 src_port мне нужно рассчитать совокупную длину пакета. Постройте ось x (относительная временная метка, которая здесь является индексом), ось y (совокупная длина пакета). Я ожидаю, что график будет содержать 3 строки для совокупной длины пакета каждого порта.

df = pd.read_csv('read.csv', sep=',')

#Calculate relative time for each dataframe
df.index = df['timestamp'] - df.loc[0,'timestamp']

#Drop unwanted columns
drop = df.drop(columns=['highest_layer', 'transport_layer','ip_flag', 'transport_flag','geo_country','data'])


df1 = drop[(drop.src_ip == '192.168.1.6') & (drop.dst_ip == '172.217.12.131')]

for i in df1['src_port']:
    df_cumsum = df1.groupby(['src_ip'])['packet_length'].cumsum()
    plt.plot(df.index, df_cumsum,label='i')

Если я укажу номера портов явно и построю его без l oop, это сработает. Но после того, как я перебираю src_port, ничего не происходит. Что мне здесь не хватает. Любые мысли, пожалуйста

Answer 1

Я создал таблицу, объединив ее с исходным DF после последующей группировки из вашего кода. Я построил график на основе этой таблицы.

df2 = df1[['src_port','packet_length']].groupby('src_port')['packet_length'].transform('cumsum').to_frame()
df2.columns = ['cumsum_packets']
df3 = pd.concat([df1,df2],axis=1)
import matplotlib.pyplot as plt
import seaborn as sns

sns.lineplot(x=df3.index, y=df3['cumsum_packets'], hue=df3['src_port'], data=df3, legend='full')

|   timestamp | src_ip      | dst_ip         |   src_port |   dst_port |   packet_length | time                       |     timestamp |   cumsum_packets |
|------------:|:------------|:---------------|-----------:|-----------:|----------------:|:---------------------------|--------------:|-----------------:|
|           0 | 192.168.1.6 | 172.217.12.131 |      32631 |         53 |              89 | 2020-06-10 19:38:08.863846 | 1591832288863 |               89 |
|           3 | 192.168.1.6 | 172.217.12.131 |      32631 |         53 |              79 | 2020-06-10 19:38:08.866492 | 1591832288866 |              168 |
|          24 | 192.168.1.6 | 172.217.12.131 |      32631 |       1900 |             216 | 2020-06-10 19:38:08.887298 | 1591832288887 |              384 |
|          25 | 192.168.1.6 | 172.217.12.131 |      32631 |        443 |              78 | 2020-06-10 19:38:08.888553 | 1591832288888 |              462 |
|          32 | 192.168.1.6 | 172.217.12.131 |      16807 |        443 |              78 | 2020-06-10 19:38:08.895594 | 1591832288895 |               78 |
|          33 | 192.168.1.6 | 172.217.12.131 |      16807 |        443 |              78 | 2020-06-10 19:38:08.896540 | 1591832288896 |              156 |
|          48 | 192.168.1.6 | 172.217.12.131 |      16807 |         53 |              75 | 2020-06-10 19:38:08.911968 | 1591832288911 |              231 |
|          53 | 192.168.1.6 | 172.217.12.131 |      58185 |        443 |              66 | 2020-06-10 19:38:08.916860 | 1591832288916 |               66 |
|          54 | 192.168.1.6 | 172.217.12.131 |      58185 |        443 |             583 | 2020-06-10 19:38:08.917442 | 1591832288917 |              649 |
|          56 | 192.168.1.6 | 172.217.12.131 |      58185 |        443 |              66 | 2020-06-10 19:38:08.919423 | 1591832288919 |              715 |
|          56 | 192.168.1.6 | 172.217.12.131 |      32631 |        443 |             583 | 2020-06-10 19:38:08.919593 | 1591832288919 |             1045 |
|          65 | 192.168.1.6 | 172.217.12.131 |      16807 |        443 |              66 | 2020-06-10 19:38:08.928922 | 1591832288928 |              297 |
|          66 | 192.168.1.6 | 172.217.12.131 |      58185 |        443 |             583 | 2020-06-10 19:38:08.929100 | 1591832288929 |             1298 |

введите описание изображения здесь