在python中将只有字符串的列表变成带有字符串和浮点数的子列表

Question

我真的在为编写 python 代码而苦苦挣扎。任何帮助将不胜感激。我需要编写代码的功能如下。由于数字是输入中的字符串，我知道我不能以某种方式按学生的姓名分开，我想我必须以某种方式将数字转换为 int/float，然后以某种方式按名称将列表分成子列表（即按字符串) 但我不知道如何做到这一点。我知道我可以使用 map(int, ) 将某些内容转换为 int，但它似乎不起作用，并且等级数（即我需要转换为浮点数）可能因任何输入而异

谢谢！

代码：

def string_list(L):

    '''(list of str) -> list of list
    Given a list of strings where each string has the format:
    'name, grade, grade, grade, ...' return a new list of 
    lists where each inner list has the format :
    [name (str), grade, grade, grade, ...] where the name 
    is a string and the grades are floats.
    >>> string_list(['Joe, 60, 90, 80', 'Harry, 60, 70', 'Jill, 98.5, 100, 95.5, 98'])
    [['Joe', 60.0, 90.0, 80.0], ['Harry', 60.0, 70.0], ['Jill', 98.5, 100.0, 95.5, 98.0]]
    '''

Answer 1

def string_list(L):
    final_list = []
    for element in L:
        new_entry = element.split(',') # Split on commas; returns a list of strings
        new_entry = [x.strip() for x in new_entry] # Remove whitespace from each string in list
        new_entry[1:] = [float(x) for x in new_entry[1:]] # Cast all entries but the first to float
        final_list.append(new_entry)
    return final_list

Answer 2

解决方案

一定要爱上神奇的 oneliner 列表理解！

[[float(val) if val.strip().isdigit() else val.strip() for val in person.split(',')] for person in L]

相同的列表理解，但适用于窄屏幕：

 [[float(val) if val.strip().isdigit() else val.strip() 
      for val in person.split(',')]  
      for person in L ]

用法

你可以像这样加入你的函数：

def string_list(L):
    return [[float(val) if val.strip().isdigit() else val.strip() for val in person.split(',')] for person in L]

然后像这样使用它：

>>> L = ["Joe, 60, 90, 80", "Harry, 60, 70", "Jill, 98.5, 100, 95.5, 98"]
>>> string_list(L)
[['Joe', 60.0, 90.0, 80.0],
 ['Harry', 60.0, 70.0],
 ['Jill', '98.5', 100.0, '95.5', 98.0]]

---

故障

列表推导很棒但是很丑！所以让我们把这个庞然大物分解成相应的部分：

for person in L]

对于我们列表中的每个项目（例如"Joe, 60, 90, 80"）

for val in person.split(',')

如果我们将这些字符串中的每一个拆分成一个字符串列表，使得 "Joe, 60, 90, 80"-> ["Joe","60","90","80"] 并存储迭代每个元素（"Joe" 或 "60" 等）将每个值存储为 val

float(val) if val.strip().isdigit() else val.strip()

如果val 仅包含数字字符 (.isdigit())，则将其转换为浮点数。否则，返回字符串（没有空格字符，如.strip()）

[[<code> for val in person.<more>] for person in L] 

内括号将人物的元素分组，没有它们你只会得到一个字符串列表。

如果您想要一个列表，您需要像这样重新排序for <thing> in <list>：

 [float(val) if val.strip().isdigit() else val.strip() 
     for person in L
     for val in person.split(',') ]