Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- params = {'rd1': 4,
- 'rd2': 2,
- 'rd3': 3}
- rows = None
- indexs = None
- columns = None
- def create_rows(params):
- '''
- Функция для создания "рядов" по заданным параметрам.
- Параметры хранятся в словаре "params".
- Функция возвращает "data" - список списков (двумерный массив).
- '''
- data = []
- for key, value in params.items():
- repeat = value
- for r in range(1, repeat+1):
- data.append([r for i in range(3)])
- return data
- rows = create_rows(params)
- indexs = [k for k, v in params.items()
- for _ in range(v)]
- columns = ['x1', 'x2', 'x3']
- # print(rows) #[[1, 1, 1], [2, 2, 2], [3, 3, 3], ...]
- # print(indexs) #['rd1', 'rd1', 'rd1', 'rd1', 'rd2', 'rd2', 'rd3', ...]
- # print(columns) #['x1', 'x2', 'x3']
- df = pd.DataFrame(rows, index=indexs, columns=columns)
- df
- # x1 x2 x3
- # rd1 1 1 1
- # rd1 2 2 2
- # rd1 3 3 3
- # rd1 4 4 4
- # rd2 1 1 1
- # rd2 2 2 2
- # rd3 1 1 1
- # rd3 2 2 2
- # rd3 3 3 3
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement