simplify pandas operations

This commit is contained in:
pemontto
2019-05-09 11:30:45 +01:00
parent 6b8e699eb6
commit 3ce6065b38
2 changed files with 2 additions and 6 deletions

View File

@ -220,11 +220,7 @@ class NessusAPI(object):
if self.profile == 'nessus':
# Set IP from asset field
df.loc[
df["asset"].str.match("^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$"), "ip"
] = df.loc[
df["asset"].str.match("^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$"), "asset"
]
df["ip"] = df.loc[df["asset"].str.match("^\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$"), "asset"]
# upper/lowercase fields
self.logger.debug('Changing case of fields')

View File

@ -524,7 +524,7 @@ class qualysScanReport:
# Extract dns field from URL
df['dns'] = df['url'].str.extract('https?://([^/]+)', expand=False)
df.loc[df['uri'] != '','dns'] = df.loc[df['uri'] != '','uri'].str.extract('https?://([^/]+)', expand=False)
df['dns'] = df.loc[df['uri'] != '','uri'].str.extract('https?://([^/]+)', expand=False)
# Set asset to web_application_name
df['asset'] = df['web_application_name']