Split array at value in numpy

后端 未结 4 1723
逝去的感伤
逝去的感伤 2020-12-10 14:10

I have a file containing data in the format:

0.0 x1
0.1 x2
0.2 x3
0.0 x4
0.1 x5
0.2 x6
0.3 x7
...

The data consists of multiple datasets, e

4条回答
  •  无人及你
    2020-12-10 14:17

    def getDataSets(fname):
        data_sets = []
        data = []
        prev = None
        with open(fname) as inf:
            for line in inf:
                index,rem = line.strip().split(None,1)
                if index < prev:
                    data_sets.append(data)
                    data = []
                data.append(rem)
                prev = index
            data_sets.append(data)
        return data_sets
    
    def main():
        data = getDataSets('split.txt')
        print data
    
    if __name__=="__main__":
        main()
    

    results in

    [['x1', 'x2', 'x3'], ['x4', 'x5', 'x6', 'x7']]
    

提交回复
热议问题