如何将数据框行转换为列?

时间:2019-08-01 18:19:57

标签: python shell

我有一个采用以下格式的数据集/数据框:

gene : ABC
sample: XYX
input:23
.
.
.
gene : DEF
sample: ERT
input :24

.
.

它持续不断。

如何以这种格式获取它?

gene sample input
abc   xyx   23
def    ert   24

.
.

任何Python或shell命令都可以。

我尝试过pd转置,但是似乎并没有给我想要的结果,  没有得到想要的输出。

1 个答案:

答案 0 :(得分:2)

我不确定100%是否在寻找什么。我将给出一些潜在解决方案的示例。如果这些与您的要求不符,请更新您的问题或添加评论。

设置(在您的示例信息之后):

    import pandas as pd
    dict1 = {"gene": "ABC", "sample": "XYZ", "input": 23}
    dict2 = {"gene": "DEF", "sample": "ERT", "input": 24}
    columns = ["gene", "sample", "input"]
    df = pd.DataFrame([dict1, dict2], columns=columns)

df的输出如下:

  gene sample  input
0  ABC    XYZ     23
1  DEF    ERT     24

看起来就像您在问题中寻找的东西。如果是这样,则可以使用类似的设置(例如开头的代码块)来设置此DataFrame。

如果您想使用该格式,并且希望对其进行转置,我将建议以下内容:

    # columns will be the index from 0 to n-1:
    df.transpose()
    # output:
    #           0    1
    # gene    ABC  DEF
    # sample  XYZ  ERT
    # input    23   24

    # try this instead
    list_that_contains_n_items_to_be_columns = ["a", "b"]
    df.index = pd.Index(list_that_contains_n_items_to_be_columns)
    df.transpose()
    # output:
    #           a    b
    # gene    ABC  DEF
    # sample  XYZ  ERT
    # input    23   24

如果您的意思是将信息发布在文本文件中,例如:

gene : ABC
sample: XYX
input:23
gene : DEF
sample: ERT
input :24

您需要将其读取并将其放入DataFrame(类似于csv格式)。您可以通过以下方式做到这一点:

import pandas as pd
list_of_dicts = []
with open("data.txt") as f:
    number_columns = 3 # change this as necessary
    line_num = 0
    for line in f:
        if line_num % number_columns == 0:
            if line_num == 0:
                dict_row = {}
            else:
                list_of_dicts.append(dict_row)
                dict_row = {}
        line_num += 1
        (key, val) = line.split(":")
        dict_row[str(key)] = val.rstrip()

# add your columns to that list
df = pd.DataFrame(list_of_dicts, columns=["gene", "sample", "input"])
print(df)

这将逐行读取文件并创建字典列表,该列表很容易变成pandas DataFrame。如果您想要一个实际的csv文件,则可以运行df.to_csv("name_of_file.csv")

希望这些帮助之一!

编辑: 要查看目录中的所有文件,可以在循环前面添加以下代码:

    import glob
    for filename in glob.glob("/your/path/here/*.txt"):
        # code you want to execute

编辑编辑:

该问题似乎与所要询问的内容无关(请参阅此答案的注释)。似乎作者拥有已经是DataFrame风格的.tsv文件,他们希望这些文件作为DataFrames读入。给出的示例文件是:

Sample Name:    1234
Index:  IB04
Input DNA:  100

Detected ITD Variants:
Size    READS   VRF



Sample Name:    1235
Index:  IB05
Input DNA:  100

Detected Variants:
Size    READS   VRF
27  112995  4.44e-01
Total   112995  4.44e-01

用于读取此文件并创建“样本” DF的示例代码:

#!/usr/bin/python
import os
import glob
import pandas as pd
os.chdir(os.getcwd())


def get_df(num_cols=3, start_key="Sample", switch_line=""):
    list_of_dfs = []
    for filepath in glob.glob("*.tsv"):
        list_of_dicts = []
        number_columns = num_cols
        line_num = 0
        part_of_df = False
        with open(filepath) as file:
            for line in file:
                # only read in lines to the df that are part of the dataframe
                if start_key in line:
                    part_of_df = True 
                elif line.strip() == "":
                    # if an empty line, go back to not adding it
                    part_of_df = False
                    continue
                if part_of_df:
                    # depending on the number of columns, add to the df
                    if line_num % number_columns == 0:
                        if line_num == 0:
                            dict_row = {}
                        else:
                            list_of_dicts.append(dict_row)
                            dict_row = {}
                    line_num += 1
                    (key, val) = line.split(":")
                    dict_row[str(key)] = val.rstrip().strip()
            if len(dict_row) % number_columns == 0:
                # if last added row is the last row of the file
                list_of_dicts.append(dict_row)
            df = pd.DataFrame(list_of_dicts, columns=['Sample Name','Index','Input DNA'])
        list_of_dfs.append(df)
    # concatenate all the files together
    final_df = pd.concat(list_of_dfs)
    return final_df

df_samples = get_df(num_cols=3, start_key="Sample", switch_line="")
print(df_samples)

这将使用基因数据创建一个DataFrame。如果这创建了您要查找的数据集,请将此答案标记为已接受。如果您还有其他问题,请提出一个新问题(在问题中发布数据文件非常有帮助)。