Как сделать namedtuple, когда в заголовках столбцов есть пробелы
Я пытаюсь создать именованный набор из объекта DictReader. Мой код выглядит так. Проблема, с которой я борюсь, заключается в том, что у меня есть очень длинные и уродливые заголовки столбцов в файле csv, с которым я работаю. В этом примере я работаю с одним из заголовков столбцов:
«Какое обычно основное блюдо на вашем ужине в День Благодарения?».
Что меня сбивает с толку, так это то, что в этом заголовке есть куча пробелов, поэтому, если я правильно понимаю, namedtuple думает, что это все аргументы. Как вы порекомендуете решить эту проблему? Я сослался на несколько потоков и чувствую, что почти добрался до них через этот: Каков питонический способ чтения данных файла CSV в виде строк именованных кортежей?
Я просто использую заголовок одного столбца в качестве примера. Вот код, который у меня есть:
import csv
import collections
filename = 'thanksgiving2015.csv'
with open(filename, 'r', encoding = 'utf-8') as f:
reader = csv.DictReader(f)
columns = collections.namedtuple('columns',
'What is typically the main dish at your
Thanksgiving dinner?')
Должен ли я удалить все эти заголовки столбцов из их пробелов перед созданием namedtuple? Я мог бы сделать это еще до того, как импортирую csv в excel, но я предполагаю, что в python есть хорошее решение.
Ответы 2
namedtuple обрабатывает одну строку как список имен полей, разделенных пробелами. Вместо этого вам нужно передать явный список имен столбцов.
namedtuple('columns', ['What is...', 'some other absurd column name'])
Однако я бы переосмыслил использование значений заголовков непосредственно в качестве имен полей. Не обращайте внимания на заголовок и передайте список более коротких имен, которые вы можете использовать в качестве атрибутов позже.
Как мне передать список более коротких имен, которые я могу использовать в качестве атрибутов позже?
Я имею в виду что-то вроде Columns = namedtuple('columns', 'foo bar bar'), чтобы позже, когда у вас будет что-то вроде x = Columns(...), вы можете записать x.foo, x.bar и т. д. x.'What is typically the main dish at your Thanksgiving dinner?' синтаксически неверен.
Да, я тоже пытался это сделать, но тоже безуспешно. Можете дать мне быстрый пример, как это сделать?
Я сделал; это самый первый фрагмент кода в моем комментарии. Выберите нужные имена, затем передайте либо список этих имен, либо одну строку, разделенную пробелами, в качестве второго аргумента.
Как отметил Чепнер, второй аргумент nametuple () может быть либо строкой, разделенной пробелами, либо списком строк, например:
columns = collections.namedtuple('columns',
['What is typically the main dish at your Thanksgiving dinner?', 'other column'])
Однако сделать это не удастся:
ValueError: Type names and field names must be valid identifiers
Это связано с тем, что columns (который вы должны использовать с заглавной буквы как Columns) будет объектом с идентификатором «Что обычно ...», а идентификаторы не могут иметь пробелов. Чтобы было ясно, вы должны использовать его как:
Columns = namedtuple('columns', ['what is', 'this'])
columns = Columns('foo', 'bar')
print(columns.this) # Works fine
print(columns.what is) # Not going to work
Если бы вы использовали простой dict (), вы бы написали:
print(columns['what is'])
Однако вы можете попросить namedtuple переименовать недопустимые идентификаторы:
Columns = namedtuple('columns', ['what is', 'this'], rename=True)
print(columns._0) # ugly but valid
print(columns.this)
Но почему вы хотите использовать namedtuple? Почему бы не прочитать список диктовок?