mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-11-20 03:54:50 +00:00
lf_csv.py: added functionality of reading csv file
This commit is contained in:
@@ -20,6 +20,7 @@ INCLUDE_IN_README
|
||||
'''
|
||||
|
||||
import pandas as pd
|
||||
from pandas import *
|
||||
|
||||
class lf_csv:
|
||||
def __init__(self,
|
||||
@@ -36,12 +37,22 @@ class lf_csv:
|
||||
|
||||
def generate_csv(self):
|
||||
df = {}
|
||||
for i in range(len(self.columns)):
|
||||
df[self.columns[i]] = self.rows[i]
|
||||
if self.rows != []:
|
||||
for i in range(len(self.columns)):
|
||||
df[self.columns[i]] = self.rows[i]
|
||||
else:
|
||||
for i in range(len(self.columns)):
|
||||
df[self.columns[i]] = []
|
||||
csv_df = pd.DataFrame(df)
|
||||
print(csv_df)
|
||||
csv_df.to_csv(self.filename, index=False, encoding='utf-8', na_rep='NA', float_format='%.2f')
|
||||
|
||||
def read_csv(self, file_name, column=None):
|
||||
data = read_csv(str(file_name))
|
||||
value = data[str(column)].tolist()
|
||||
print("value of column", value)
|
||||
return value
|
||||
|
||||
def main():
|
||||
test = lf_csv()
|
||||
test.generate_csv()
|
||||
|
||||
Reference in New Issue
Block a user